1*523fa7a6SAndroid Build Coastguard Worker# @noautodeps 2*523fa7a6SAndroid Build Coastguard Worker 3*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:python_library.bzl", "python_library") 4*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 5*523fa7a6SAndroid Build Coastguard Worker 6*523fa7a6SAndroid Build Coastguard Workeroncall("executorch") 7*523fa7a6SAndroid Build Coastguard Worker 8*523fa7a6SAndroid Build Coastguard Workerpython_library( 9*523fa7a6SAndroid Build Coastguard Worker name = "exported_module", 10*523fa7a6SAndroid Build Coastguard Worker srcs = [ 11*523fa7a6SAndroid Build Coastguard Worker "exported_module.py", 12*523fa7a6SAndroid Build Coastguard Worker ], 13*523fa7a6SAndroid Build Coastguard Worker visibility = [ 14*523fa7a6SAndroid Build Coastguard Worker "//executorch/...", 15*523fa7a6SAndroid Build Coastguard Worker ], 16*523fa7a6SAndroid Build Coastguard Worker deps = [ 17*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 18*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:dynamic_shape", 19*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 20*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_manager", 21*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 22*523fa7a6SAndroid Build Coastguard Worker ], 23*523fa7a6SAndroid Build Coastguard Worker) 24*523fa7a6SAndroid Build Coastguard Worker 25*523fa7a6SAndroid Build Coastguard Workerpython_library( 26*523fa7a6SAndroid Build Coastguard Worker name = "register_scratch_meta_fns", 27*523fa7a6SAndroid Build Coastguard Worker srcs = [ 28*523fa7a6SAndroid Build Coastguard Worker "register_scratch_meta_fns.py", 29*523fa7a6SAndroid Build Coastguard Worker ], 30*523fa7a6SAndroid Build Coastguard Worker deps = [ 31*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/operator:manip", 32*523fa7a6SAndroid Build Coastguard Worker ], 33*523fa7a6SAndroid Build Coastguard Worker) 34*523fa7a6SAndroid Build Coastguard Worker 35*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 36*523fa7a6SAndroid Build Coastguard Worker name = "end2end_aten", 37*523fa7a6SAndroid Build Coastguard Worker srcs = [ 38*523fa7a6SAndroid Build Coastguard Worker "test_end2end.py", 39*523fa7a6SAndroid Build Coastguard Worker ], 40*523fa7a6SAndroid Build Coastguard Worker preload_deps = ["//executorch/kernels/portable:custom_ops_generated_lib"], 41*523fa7a6SAndroid Build Coastguard Worker deps = [ 42*523fa7a6SAndroid Build Coastguard Worker ":exported_module", 43*523fa7a6SAndroid Build Coastguard Worker ":register_scratch_meta_fns", 44*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 45*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools:lib", 46*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools/bundled_program:config", 47*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools/bundled_program/serialize:lib", 48*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:dynamic_shape", 49*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 50*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:memory", 51*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_manager", 52*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:print_program", 53*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:tensor", 54*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/_serialize:lib", 55*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/emit:lib", 56*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 57*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:control_flow_models", 58*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:dynamic_shape_models", 59*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:models", 60*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:transformer", 61*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/pybindings:aten_lib", 62*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/pytree:pybindings", 63*523fa7a6SAndroid Build Coastguard Worker ], 64*523fa7a6SAndroid Build Coastguard Worker) 65*523fa7a6SAndroid Build Coastguard Worker 66*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 67*523fa7a6SAndroid Build Coastguard Worker name = "end2end", 68*523fa7a6SAndroid Build Coastguard Worker srcs = [ 69*523fa7a6SAndroid Build Coastguard Worker "test_end2end.py", 70*523fa7a6SAndroid Build Coastguard Worker ], 71*523fa7a6SAndroid Build Coastguard Worker preload_deps = ["//executorch/kernels/portable:custom_ops_generated_lib"], 72*523fa7a6SAndroid Build Coastguard Worker deps = [ 73*523fa7a6SAndroid Build Coastguard Worker ":exported_module", 74*523fa7a6SAndroid Build Coastguard Worker ":register_scratch_meta_fns", 75*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 76*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools:lib", 77*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools/bundled_program:config", 78*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools/bundled_program/serialize:lib", 79*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:dynamic_shape", 80*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 81*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:memory", 82*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_manager", 83*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:print_program", 84*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:tensor", 85*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/_serialize:lib", 86*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/emit:lib", 87*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 88*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:control_flow_models", 89*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:dynamic_shape_models", 90*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:models", 91*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/tests:transformer", 92*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/pybindings:portable_lib", 93*523fa7a6SAndroid Build Coastguard Worker "//executorch/extension/pytree:pybindings", 94*523fa7a6SAndroid Build Coastguard Worker ], 95*523fa7a6SAndroid Build Coastguard Worker) 96