We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f3c92ae commit 0dfda64Copy full SHA for 0dfda64
backends/cuda/TARGETS
@@ -2,6 +2,20 @@ load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2
3
oncall("executorch")
4
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
+
19
runtime.python_library(
20
name = "cuda_backend",
21
srcs = [
@@ -11,6 +25,7 @@ runtime.python_library(
25
"//executorch/...",
26
],
27
deps = [
28
+ ":cuda_passes",
29
":triton_replacement_pass",
30
"//caffe2:torch",
31
"//executorch/backends/aoti/passes:passes",
0 commit comments