diff --git a/backends/cuda/TARGETS b/backends/cuda/TARGETS index 3ae4eec6680..3892d5e1742 100644 --- a/backends/cuda/TARGETS +++ b/backends/cuda/TARGETS @@ -2,6 +2,20 @@ load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") oncall("executorch") +runtime.python_library( + name = "cuda_passes", + srcs = [ + "passes/__init__.py", + "passes/move_cond_predicate_to_cpu.py", + ], + visibility = [ + "//executorch/backends/cuda/...", + ], + deps = [ + "//caffe2:torch", + ], +) + runtime.python_library( name = "cuda_backend", srcs = [ @@ -11,6 +25,7 @@ runtime.python_library( "//executorch/...", ], deps = [ + ":cuda_passes", ":triton_replacement_pass", "//caffe2:torch", "//executorch/backends/aoti/passes:passes",