New compute at interface (csarofeen#1743)

Rewrite of the compute at pass to rely on the new propagation mechanisms.
shmsong · Jul 1, 2022 · 28cbaf9 · 28cbaf9
1 parent 635ebfc
commit 28cbaf9
Show file tree

Hide file tree

Showing 6 changed files with 677 additions and 812 deletions.
diff --git a/build_variables.bzl b/build_variables.bzl
@@ -642,6 +642,7 @@ libtorch_cuda_core_sources = [
     "torch/csrc/autograd/functions/comm.cpp",
     "torch/csrc/jit/codegen/cuda/arith.cpp",
     "torch/csrc/jit/codegen/cuda/compute_at.cpp",
+    "torch/csrc/jit/codegen/cuda/inline_propagator.cpp",
     "torch/csrc/jit/codegen/cuda/compute_at_map.cpp",
     "torch/csrc/jit/codegen/cuda/codegen.cpp",
     "torch/csrc/jit/codegen/cuda/contiguity.cpp",