pytorch
7e2a7728 - dbr quant: record dag of non-quantizeable ops (#71551)

Commit
2 years ago
dbr quant: record dag of non-quantizeable ops (#71551) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71551 This PR makes DBR quant record the DAG of non-quantizeable ops. Having this will enable us to analyze the entire traced graph of pytorch ops, regardless of whether they support quantization or not. That, in turn, will enable analysis of uses of each op, allowing us to safely determine whether a subgraph of ops can be fused or not. In future PRs, this functionality will be used to implement function fusion. Test Plan: ``` python test/test_quantization.py -k DBR ``` Reviewed By: jerryzh168 Differential Revision: D33684130 Pulled By: vkuzo fbshipit-source-id: 497d9882f0670a36eef2a0900ea2517c82addf66 (cherry picked from commit b0d48a8e66a42213b353458fc86b2eb537f43828)
Author
Committer
Parents
Loading