OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ad78edee8e
pytorch/torch/ao/quantization
History
Jerry Zhang ad78edee8e Add support for list, tuple and dict in numeric debugger (#143882) 
Summary:
Previously numeric debugger only supports torch.Tensor, this PR adds support for list, tuple and dict as well

Test Plan:
python test/test_quantization.py -k test_extract_results_from_loggers_list_output

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D67660049](https://our.internmc.facebook.com/intern/diff/D67660049)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143882
Approved by: https://github.com/dulinriley
2024-12-28 02:10:31 +00:00
..
backend_config Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
experimental remove allow-untyped-defs from torch/ao/quantization/experimental/APoT_tensor.py (#143601) 2024-12-20 05:26:09 +00:00
fx [Quantization] add an option keep_original_weights in _lower_to_native_backend (#141049) 2024-12-27 04:02:07 +00:00
pt2e Add support for list, tuple and dict in numeric debugger (#143882) 2024-12-28 02:10:31 +00:00
quantizer [reland] Kill capture_pre_autograd_graph API (#143426) 2024-12-18 12:07:09 +00:00
__init__.py Add support for prototype affine quantization in pt2e flow (#141421) 2024-12-24 04:22:18 +00:00
_correct_bias.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
_equalize.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
_learnable_fake_quantize.py [BE] enable UFMT for torch/ao/quantization/ (#128863) 2024-07-25 04:17:54 +00:00
fake_quantize.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
fuse_modules.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
fuser_method_mappings.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
observer.py Add support for prototype affine quantization in pt2e flow (#141421) 2024-12-24 04:22:18 +00:00
pattern.md
qconfig_mapping.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
qconfig.py Revise CPU vectorization ISA support API (#135075) 2024-09-05 12:14:56 +00:00
quant_type.py [BE] enable UFMT for torch/ao/quantization/ (#128863) 2024-07-25 04:17:54 +00:00
quantization_mappings.py [BE] enable UFMT for torch/ao/quantization/ (#128863) 2024-07-25 04:17:54 +00:00
quantize_fx.py [Quantization] add an option keep_original_weights in _lower_to_native_backend (#141049) 2024-12-27 04:02:07 +00:00
quantize_jit.py [BE] enable UFMT for torch/ao/quantization/ (#128863) 2024-07-25 04:17:54 +00:00
quantize_pt2e.py [BC-Breaking]Remove capture_pre_autograd_graph references in quantization (#139505) 2024-12-13 22:26:22 +00:00
quantize.py Revert "Handle meta tensors in FX quantization (#142262)" 2024-12-21 20:34:09 +00:00
stubs.py [BE] enable UFMT for torch/ao/quantization/ (#128863) 2024-07-25 04:17:54 +00:00
utils.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00