pytorch
76764330 - [AOTInductor] Reuse generated kernels between constant graph and main graph (#121564)

Commit
315 days ago
[AOTInductor] Reuse generated kernels between constant graph and main graph (#121564) Summary: We copy the src_to_kernel from constant graph to main graph so that we could avoid generating duplicating kernels. And pass throught the name counter such that no duplicated names will be generated. Test Plan: Included in commit Differential Revision: D54706767 Pull Request resolved: https://github.com/pytorch/pytorch/pull/121564 Approved by: https://github.com/desertfire, https://github.com/chenyang78
Author
Committer
Parents
Loading