pytorch
93de8020 - ns for fx: move node I/O dtype mapping to be local instead of global (#56296)

Commit
3 years ago
ns for fx: move node I/O dtype mapping to be local instead of global (#56296) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56296 To support shadows of custom functions, we need to allow user to specify I/O type of the custom functions. This PR is a cleanup in preparation for making the above happen. We make the I/O dtype mappings be generated by a function instead of a global variable. In the next PR, we will add a hook so user can modify these mappings. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27831996 fbshipit-source-id: 782f5e77de0eef3899b9b7def0fdabd8dcafef12
Author
Parents
Loading