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