1# @noautodeps 2 3load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 4 5oncall("executorch") 6 7python_unittest( 8 name = "bundle_data", 9 srcs = [ 10 "test_bundle_data.py", 11 ], 12 deps = [ 13 "//caffe2:torch", 14 "//executorch/devtools/bundled_program:config", 15 "//executorch/devtools/bundled_program:core", 16 "//executorch/devtools/bundled_program/schema:bundled_program_schema_py", 17 "//executorch/devtools/bundled_program/util:test_util", 18 "//executorch/exir/_serialize:lib", 19 ], 20) 21 22python_unittest( 23 name = "config", 24 srcs = [ 25 "test_config.py", 26 ], 27 deps = [ 28 "//caffe2:torch", 29 "//executorch/devtools/bundled_program:config", 30 "//executorch/devtools/bundled_program/util:test_util", 31 "//executorch/extension/pytree:pylib", 32 ], 33) 34 35python_unittest( 36 name = "end2end", 37 srcs = [ 38 "test_end2end.py", 39 ], 40 deps = [ 41 "//caffe2:torch", 42 "//executorch/devtools/bundled_program:config", 43 "//executorch/devtools/bundled_program:core", 44 "//executorch/devtools/bundled_program/serialize:lib", 45 "//executorch/devtools/bundled_program/util:test_util", 46 "//executorch/exir:dynamic_shape", 47 "//executorch/exir:lib", 48 "//executorch/exir:memory", 49 "//executorch/exir:pass_manager", 50 "//executorch/exir:print_program", 51 "//executorch/exir:tensor", 52 "//executorch/exir/_serialize:lib", 53 "//executorch/exir/emit:lib", 54 "//executorch/exir/passes:lib", 55 "//executorch/exir/tests:control_flow_models", 56 "//executorch/exir/tests:dynamic_shape_models", 57 "//executorch/exir/tests:models", 58 "//executorch/exir/tests:transformer", 59 "//executorch/extension/pybindings:portable_lib", 60 "//executorch/extension/pytree:pybindings", 61 "//executorch/kernels/portable:custom_ops_generated_lib", 62 ], 63) 64