blob: 652c74b8f43ddb20d15c4464b8eb9541a9fe6afb [file] [log] [blame]
# @noautodeps
load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest")
oncall("executorch")
python_unittest(
name = "bundle_data",
srcs = [
"test_bundle_data.py",
],
deps = [
"//caffe2:torch",
"//executorch/devtools/bundled_program:config",
"//executorch/devtools/bundled_program:core",
"//executorch/devtools/bundled_program/schema:bundled_program_schema_py",
"//executorch/devtools/bundled_program/util:test_util",
"//executorch/exir/_serialize:lib",
],
)
python_unittest(
name = "config",
srcs = [
"test_config.py",
],
deps = [
"//caffe2:torch",
"//executorch/devtools/bundled_program:config",
"//executorch/devtools/bundled_program/util:test_util",
"//executorch/extension/pytree:pylib",
],
)
python_unittest(
name = "end2end",
srcs = [
"test_end2end.py",
],
deps = [
"//caffe2:torch",
"//executorch/devtools/bundled_program:config",
"//executorch/devtools/bundled_program:core",
"//executorch/devtools/bundled_program/serialize:lib",
"//executorch/devtools/bundled_program/util:test_util",
"//executorch/exir:dynamic_shape",
"//executorch/exir:lib",
"//executorch/exir:memory",
"//executorch/exir:pass_manager",
"//executorch/exir:print_program",
"//executorch/exir:tensor",
"//executorch/exir/_serialize:lib",
"//executorch/exir/emit:lib",
"//executorch/exir/passes:lib",
"//executorch/exir/tests:control_flow_models",
"//executorch/exir/tests:dynamic_shape_models",
"//executorch/exir/tests:models",
"//executorch/exir/tests:transformer",
"//executorch/extension/pybindings:portable_lib",
"//executorch/extension/pytree:pybindings",
"//executorch/kernels/portable:custom_ops_generated_lib",
],
)