pytorch
9c455d70 - dbr quant: add limited support for `torch.nn.ModuleList` (#70372)

Commit
2 years ago
dbr quant: add limited support for `torch.nn.ModuleList` (#70372) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70372 Enables basic support for `torch.nn.ModuleList` in DBR quant by stopping it from being a leaf. For now, we require the user to check for `AutoQuantizationState` if they are looping over the contents without any bounds checking. In future PRs, we can explore how to solve this without requiring user code changes. Test Plan: ``` python test/test_quantization.py TestQuantizeDBR.test_module_list ``` Reviewed By: VitalyFedyunin Differential Revision: D33302329 Pulled By: vkuzo fbshipit-source-id: 1604748d4b6c2b9d14b50df46268246da807d539
Author
Parents
Loading