1load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 2load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") 3 4oncall("executorch") 5 6runtime.python_library( 7 name = "insert_prepack_nodes", 8 srcs = ["insert_prepack_nodes.py"], 9 visibility = [ 10 "//executorch/backends/...", 11 ], 12 deps = [ 13 "//caffe2:torch", 14 "//executorch/exir:pass_base", 15 "//executorch/backends/vulkan:utils_lib", 16 ], 17) 18 19runtime.python_library( 20 name = "int4_weight_only_quantizer", 21 srcs = [ 22 "int4_weight_only_quantizer.py", 23 ], 24 visibility = [ 25 "//executorch/backends/...", 26 ], 27 deps = [ 28 "//executorch/backends/vulkan:custom_ops_lib", 29 "//pytorch/ao:torchao", 30 ] 31) 32 33runtime.python_library( 34 name = "remove_local_scalar_dense", 35 srcs = ["remove_local_scalar_dense_ops.py"], 36 visibility = [ 37 "//executorch/backends/...", 38 ], 39 deps = [ 40 "//caffe2:torch", 41 "//executorch/exir:pass_base", 42 "//executorch/exir/dialects:lib", 43 ], 44) 45 46runtime.python_library( 47 name = "tag_memory_meta_pass", 48 srcs = ["tag_memory_meta_pass.py"], 49 visibility = [ 50 "//executorch/backends/...", 51 ], 52 deps = [ 53 "//caffe2:torch", 54 "//executorch/exir:pass_base", 55 "//executorch/exir/dialects:lib", 56 "//executorch/backends/vulkan:utils_lib", 57 "//executorch/backends/vulkan/serialization:lib", 58 ], 59) 60 61runtime.python_library( 62 name = "vulkan_passes", 63 srcs = [ 64 "__init__.py", 65 ], 66 visibility = [ 67 "//executorch/backends/...", 68 "//executorch/examples/...", 69 ], 70 deps = [ 71 ":insert_prepack_nodes", 72 ":int4_weight_only_quantizer", 73 ":remove_local_scalar_dense", 74 ":tag_memory_meta_pass" 75 ] 76) 77