File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments