pytorch
a2e809f2 - [quant][pt2e] Fix a bug in reference quantized module (decomposed mode) (#98903)

Comment changes are shownComment changes are hidden
Commit
2 years ago
[quant][pt2e] Fix a bug in reference quantized module (decomposed mode) (#98903) Summary: Fixed quant_min/quant_max for per channel quantized weight for reference quantized module in decomposed mode, this bug is triggered while onboard an internal model Test Plan: python test/test_quantization.py TestQuantizeFx.test__convert_to_reference_decomposed_fx_per_channel_quant_module Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/98903 Approved by: https://github.com/andrewor14
Author
Committer
Parents
  • test/quantization/fx
    • File
      test_quantize_fx.py
  • torch/ao/nn/quantized/reference/modules
    • File
      utils.py
Loading