Skip to content

Commit 0dfda64

Browse files
authored
CUDA backend: fix targets file for passes
Differential Revision: D89825579 Pull Request resolved: #16401
1 parent f3c92ae commit 0dfda64

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

backends/cuda/TARGETS

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,20 @@ load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
22

33
oncall("executorch")
44

5+
runtime.python_library(
6+
name = "cuda_passes",
7+
srcs = [
8+
"passes/__init__.py",
9+
"passes/move_cond_predicate_to_cpu.py",
10+
],
11+
visibility = [
12+
"//executorch/backends/cuda/...",
13+
],
14+
deps = [
15+
"//caffe2:torch",
16+
],
17+
)
18+
519
runtime.python_library(
620
name = "cuda_backend",
721
srcs = [
@@ -11,6 +25,7 @@ runtime.python_library(
1125
"//executorch/...",
1226
],
1327
deps = [
28+
":cuda_passes",
1429
":triton_replacement_pass",
1530
"//caffe2:torch",
1631
"//executorch/backends/aoti/passes:passes",

0 commit comments

Comments
 (0)