pytorch
16554bec - [qunat][fx][fix] Fix get_module_type for fusion (#72735)

Commit
2 years ago
[qunat][fx][fix] Fix get_module_type for fusion (#72735) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72735 We use `get_matched_types` to get the (type) pattern from matched modules. And we need to use MatchAllNode instead of type(MatchAllNode) to query the fuser_method for the pattern Test Plan: TODO Imported from OSS Reviewed By: raghuramank10000 Differential Revision: D34180705 fbshipit-source-id: db9b6e791a9f26b70079fddc95fce033052199ab (cherry picked from commit 01d38afabcb1bfc207dee7d49ee13df500d32fdf)
Author
Committer
Parents
Loading