pytorch
cb4b3b04 - [nnc] Move device type checks from isSupported to typesAreSupported (#55025)

Commit
3 years ago
[nnc] Move device type checks from isSupported to typesAreSupported (#55025) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55025 Something needs to be fixed about the names of these functions, because they are confusing. The profiling infrastructure calls `isSupported` to see if it should insert profiling nodes. The fuser calls `isSupported` but also `typesAreSupported` to determine if it can actually fuse the node. At profiling time, we don't know device types yet, so we can't use device type checks in `isSupported` or else we'll never profile the node. So we want to move those checks into `typesAreSupported`, where we actually have profiling info available. ghstack-source-id: 126076111 Test Plan: sandcastle Reviewed By: ngimel Differential Revision: D27454968 fbshipit-source-id: 4ffb142ea7a0086842a034c9e202f9cb1065fc95
Author
Parents
Loading