1*523fa7a6SAndroid Build Coastguard Workerload("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") 2*523fa7a6SAndroid Build Coastguard Workerload(":targets.bzl", "define_common_targets") 3*523fa7a6SAndroid Build Coastguard Worker 4*523fa7a6SAndroid Build Coastguard Workeroncall("executorch") 5*523fa7a6SAndroid Build Coastguard Worker 6*523fa7a6SAndroid Build Coastguard Workerdefine_common_targets() 7*523fa7a6SAndroid Build Coastguard Worker 8*523fa7a6SAndroid Build Coastguard Workerruntime.python_test( 9*523fa7a6SAndroid Build Coastguard Worker name = "test_xnnpack_passes", 10*523fa7a6SAndroid Build Coastguard Worker srcs = glob([ 11*523fa7a6SAndroid Build Coastguard Worker "passes/*.py", 12*523fa7a6SAndroid Build Coastguard Worker ]) + [ 13*523fa7a6SAndroid Build Coastguard Worker "test_xnnpack_utils_classes.py", 14*523fa7a6SAndroid Build Coastguard Worker ], 15*523fa7a6SAndroid Build Coastguard Worker deps = [ 16*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack/_passes:xnnpack_passes", 17*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack/test/tester:tester", 18*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack/utils:xnnpack_utils", 19*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 20*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 21*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/backend/canonical_partitioners:canonical_partitioner_lib", 22*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 23*523fa7a6SAndroid Build Coastguard Worker ], 24*523fa7a6SAndroid Build Coastguard Worker external_deps = [ 25*523fa7a6SAndroid Build Coastguard Worker "libtorch", 26*523fa7a6SAndroid Build Coastguard Worker ], 27*523fa7a6SAndroid Build Coastguard Worker) 28*523fa7a6SAndroid Build Coastguard Worker 29*523fa7a6SAndroid Build Coastguard Workerruntime.python_test( 30*523fa7a6SAndroid Build Coastguard Worker name = "test_xnnpack_ops", 31*523fa7a6SAndroid Build Coastguard Worker srcs = glob([ 32*523fa7a6SAndroid Build Coastguard Worker "ops/*.py", 33*523fa7a6SAndroid Build Coastguard Worker ]) + [ 34*523fa7a6SAndroid Build Coastguard Worker "test_xnnpack_utils.py", 35*523fa7a6SAndroid Build Coastguard Worker ], 36*523fa7a6SAndroid Build Coastguard Worker deps = [ 37*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack/partition:xnnpack_partitioner", 38*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack/test/tester:tester", 39*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools:lib", 40*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools/bundled_program:config", 41*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools/bundled_program/serialize:lib", 42*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:constant_prop_pass", 43*523fa7a6SAndroid Build Coastguard Worker "//pytorch/ao:torchao", # @manual 44*523fa7a6SAndroid Build Coastguard Worker ], 45*523fa7a6SAndroid Build Coastguard Worker external_deps = [ 46*523fa7a6SAndroid Build Coastguard Worker "libtorch", 47*523fa7a6SAndroid Build Coastguard Worker ], 48*523fa7a6SAndroid Build Coastguard Worker) 49*523fa7a6SAndroid Build Coastguard Worker 50*523fa7a6SAndroid Build Coastguard Workerruntime.python_test( 51*523fa7a6SAndroid Build Coastguard Worker name = "test_xnnpack_models", 52*523fa7a6SAndroid Build Coastguard Worker srcs = glob([ 53*523fa7a6SAndroid Build Coastguard Worker "models/*.py", 54*523fa7a6SAndroid Build Coastguard Worker ]), 55*523fa7a6SAndroid Build Coastguard Worker tags = ["long_running"], 56*523fa7a6SAndroid Build Coastguard Worker deps = [ 57*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/timm:timm", 58*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/torchsr:torchsr", # @manual 59*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/transformers:transformers", # @manual 60*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack/test/tester:tester", 61*523fa7a6SAndroid Build Coastguard Worker "//executorch/examples/models/llama:llama2_model", 62*523fa7a6SAndroid Build Coastguard Worker "//pytorch/audio/src:torchaudio_core", 63*523fa7a6SAndroid Build Coastguard Worker "//pytorch/vision:torchvision", # @manual 64*523fa7a6SAndroid Build Coastguard Worker ], 65*523fa7a6SAndroid Build Coastguard Worker external_deps = [ 66*523fa7a6SAndroid Build Coastguard Worker "libtorch", 67*523fa7a6SAndroid Build Coastguard Worker ], 68*523fa7a6SAndroid Build Coastguard Worker) 69*523fa7a6SAndroid Build Coastguard Worker 70*523fa7a6SAndroid Build Coastguard Workerruntime.python_test( 71*523fa7a6SAndroid Build Coastguard Worker name = "test_xnnpack_serialization", 72*523fa7a6SAndroid Build Coastguard Worker srcs = glob([ 73*523fa7a6SAndroid Build Coastguard Worker "serialization/*.py", 74*523fa7a6SAndroid Build Coastguard Worker ]), 75*523fa7a6SAndroid Build Coastguard Worker deps = [ 76*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/xnnpack:xnnpack_preprocess", 77*523fa7a6SAndroid Build Coastguard Worker ], 78*523fa7a6SAndroid Build Coastguard Worker) 79