blob: bbddb70735c072703ff8837196f947384ef6de8d [file] [edit]
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",
],
)