1load("@fbcode_macros//build_defs:python_library.bzl", "python_library") 2load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 3 4oncall("executorch") 5 6python_library( 7 name = "example_quantizer", 8 srcs = [ 9 "example_quantizer.py", 10 ], 11 deps = [ 12 "//caffe2:torch", 13 "//executorch/backends/example/example_operators:example_operators_lib", 14 ], 15) 16 17python_library( 18 name = "example_backend", 19 srcs = [ 20 "example_backend.py", 21 ], 22 deps = [ 23 "//executorch/backends/example/example_backend_delegate_passes:lib", 24 "//executorch/exir/backend:backend_details", 25 "//executorch/exir/backend:compile_spec_schema", 26 ], 27) 28 29python_library( 30 name = "example_partitioner", 31 srcs = [ 32 "example_partitioner.py", 33 ], 34 deps = [ 35 ":example_backend", 36 "//caffe2:torch", 37 "//executorch/backends/example/example_operators:example_operators_lib", 38 "//executorch/exir:graph_module", 39 "//executorch/exir/backend:partitioner", 40 "//executorch/exir/backend/canonical_partitioners:canonical_partitioner_lib", 41 "//executorch/exir/dialects:lib", 42 ], 43) 44 45python_unittest( 46 name = "test_example_delegate", 47 srcs = [ 48 "test_example_delegate.py", 49 ], 50 deps = [ 51 ":example_partitioner", 52 ":example_quantizer", 53 "//caffe2:torch", 54 "//executorch/exir:delegate", 55 "//executorch/exir:lib", 56 "//executorch/exir/backend/canonical_partitioners:canonical_partitioner_lib", 57 "//pytorch/vision:torchvision", 58 ], 59) 60