saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/quantization
History
Jerry Zhang 3ca71ed735 [quant][pt2e] store scale/zero_point as tensor attributes to support serialization (#105894) 
Summary:
Currently scale/zero_point for per tensor quant is stored as burnt in literals, this means these values can't be serialized in state_dict, this
PR changes them to buffers/Tensors so that they can be serialized

Test Plan:
python test/test_quantization.py TestQuantizePT2E

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D47770963](https://our.internmc.facebook.com/intern/diff/D47770963)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105894
Approved by: https://github.com/kimishpatel
2023-07-26 20:15:06 +00:00
..
ao_migration
bc [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
core add basic cuda support for float8 dtypes (#105807) 2023-07-25 03:43:36 +00:00
eager [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
fx [quant][pt2e] store scale/zero_point as tensor attributes to support serialization (#105894) 2023-07-26 20:15:06 +00:00
jit [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
pt2e [quant][pt2e] store scale/zero_point as tensor attributes to support serialization (#105894) 2023-07-26 20:15:06 +00:00
serialized
__init__.py