| load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") |
| load(":targets.bzl", "define_common_targets") |
| |
| oncall("executorch") |
| |
| define_common_targets(is_fbcode = True) |
| |
| runtime.python_library( |
| name = "vulkan_preprocess", |
| srcs = [ |
| "serialization/vulkan_graph_builder.py", |
| "serialization/vulkan_graph_schema.py", |
| "serialization/vulkan_graph_serialize.py", |
| "vulkan_preprocess.py", |
| ], |
| resources = [ |
| "serialization/schema.fbs", |
| ], |
| visibility = [ |
| "//executorch/...", |
| "//executorch/vulkan/...", |
| "@EXECUTORCH_CLIENTS", |
| ], |
| deps = [ |
| "//executorch/backends/transforms:addmm_mm_to_linear", |
| "//executorch/backends/transforms:fuse_batch_norm_with_conv", |
| "//executorch/backends/transforms:fuse_conv_with_clamp", |
| "//executorch/backends/transforms:fuse_view_copy", |
| "//executorch/backends/transforms:mean_to_sum_div", |
| "//executorch/backends/transforms:remove_clone_ops", |
| "//executorch/backends/vulkan/_passes:remove_local_scalar_dense", |
| "//executorch/exir:graph_module", |
| "//executorch/exir/_serialize:_bindings", |
| "//executorch/exir/_serialize:lib", |
| "//executorch/exir/backend:backend_details", |
| ], |
| ) |