frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[LT] Move MakeNode into ir_builder.h

Browse Source 
Summary: Move MakeNode into ir_builder.h to avoid circular header
reference later when introducing a trie cache for IR node lookup.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76482

Approved by: https://github.com/wconstab
This commit is contained in:
Bin Bao
2022-05-03 00:54:20 +00:00
committed by PyTorch MergeBot
parent 138c944a6b
commit f8a4780eb2
8 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torchgen/gen_lazy_tensor.py
View File

@ -451,6 +451,7 @@ def run_gen_lazy_tensor(
"ATen/MetaFunctions.h",
"ATen/Operators.h",
"ATen/native/CPUFallback.h",
"torch/csrc/lazy/core/ir_builder.h",
"torch/csrc/lazy/core/lazy_graph_executor.h",
"torch/csrc/lazy/core/metrics.h",
"torch/csrc/lazy/core/shape.h",