Skip to content

Commit 52830fe

Browse files
committed
Arm backend: Package VGF custom shader passes
Change-Id: Iae92c23da049949f5f1d0719922d84581893d4f7
1 parent 76e6e22 commit 52830fe

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

backends/arm/TARGETS

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,15 +87,27 @@ runtime.python_library(
8787
name = "vgf",
8888
srcs = [
8989
"vgf/__init__.py",
90+
"vgf/_passes/__init__.py",
91+
"vgf/_passes/rewrite_grid_sampler_to_tosa_custom.py",
9092
"vgf/backend.py",
9193
"vgf/compile_spec.py",
9294
"vgf/model_converter.py",
9395
"vgf/partitioner.py",
96+
"vgf/shaders/__init__.py",
97+
"vgf/shaders/grid_sampler.py",
98+
],
99+
resources = [
100+
"vgf/shaders/grid_sampler.glsl",
101+
"vgf/shaders/grid_sampler.spirv.b64",
94102
],
95103
deps = [
96104
":arm_compile_spec",
105+
"//caffe2:torch",
106+
"//executorch/backends/arm/_passes:passes",
107+
"//executorch/backends/arm/tosa/dialect:lib",
97108
"//executorch/backends/arm/tosa:specification",
98109
"//executorch/backends/arm/tosa:partitioner",
110+
"//executorch/exir:lib",
99111
],
100112
)
101113

0 commit comments

Comments
 (0)