Skip to content

Commit 28cbaf9

Browse files
authored
New compute at interface (#1743)
Rewrite of the compute at pass to rely on the new propagation mechanisms.
1 parent 635ebfc commit 28cbaf9

File tree

6 files changed

+677
-812
lines changed

6 files changed

+677
-812
lines changed

build_variables.bzl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -642,6 +642,7 @@ libtorch_cuda_core_sources = [
642642
"torch/csrc/autograd/functions/comm.cpp",
643643
"torch/csrc/jit/codegen/cuda/arith.cpp",
644644
"torch/csrc/jit/codegen/cuda/compute_at.cpp",
645+
"torch/csrc/jit/codegen/cuda/inline_propagator.cpp",
645646
"torch/csrc/jit/codegen/cuda/compute_at_map.cpp",
646647
"torch/csrc/jit/codegen/cuda/codegen.cpp",
647648
"torch/csrc/jit/codegen/cuda/contiguity.cpp",

0 commit comments

Comments
 (0)