1*523fa7a6SAndroid Build Coastguard Worker# @noautodeps 2*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:python_library.bzl", "python_library") 3*523fa7a6SAndroid Build Coastguard Worker 4*523fa7a6SAndroid Build Coastguard Workerpython_library( 5*523fa7a6SAndroid Build Coastguard Worker name = "arm_partitioner", 6*523fa7a6SAndroid Build Coastguard Worker srcs = [ 7*523fa7a6SAndroid Build Coastguard Worker "arm_partitioner.py", 8*523fa7a6SAndroid Build Coastguard Worker ], 9*523fa7a6SAndroid Build Coastguard Worker typing = True, 10*523fa7a6SAndroid Build Coastguard Worker deps = [ 11*523fa7a6SAndroid Build Coastguard Worker ":arm_backend", 12*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/operator_support:operator_support", 13*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/_passes:passes", 14*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 15*523fa7a6SAndroid Build Coastguard Worker ], 16*523fa7a6SAndroid Build Coastguard Worker) 17*523fa7a6SAndroid Build Coastguard Worker 18*523fa7a6SAndroid Build Coastguard Workerpython_library( 19*523fa7a6SAndroid Build Coastguard Worker name = "arm_backend", 20*523fa7a6SAndroid Build Coastguard Worker srcs = [ 21*523fa7a6SAndroid Build Coastguard Worker "arm_backend.py", 22*523fa7a6SAndroid Build Coastguard Worker ], 23*523fa7a6SAndroid Build Coastguard Worker typing = True, 24*523fa7a6SAndroid Build Coastguard Worker deps = [ 25*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/flatbuffers:flatbuffers", 26*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/ml-dtypes:ml-dtypes", 27*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/serializer:serializer", 28*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/tosa:tosa", 29*523fa7a6SAndroid Build Coastguard Worker ":arm_vela", 30*523fa7a6SAndroid Build Coastguard Worker ":process_node", 31*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/operators:lib", 32*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/operators:node_visitor", 33*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/_passes:passes", 34*523fa7a6SAndroid Build Coastguard Worker ], 35*523fa7a6SAndroid Build Coastguard Worker) 36*523fa7a6SAndroid Build Coastguard Worker 37*523fa7a6SAndroid Build Coastguard Workerpython_library( 38*523fa7a6SAndroid Build Coastguard Worker name = "process_node", 39*523fa7a6SAndroid Build Coastguard Worker srcs = ["process_node.py"], 40*523fa7a6SAndroid Build Coastguard Worker typing = True, 41*523fa7a6SAndroid Build Coastguard Worker deps = [ 42*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/tosa:tosa", 43*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/operators:node_visitor", 44*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm:tosa_mapping", 45*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm:tosa_quant_utils", 46*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm:tosa_utils", 47*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 48*523fa7a6SAndroid Build Coastguard Worker ], 49*523fa7a6SAndroid Build Coastguard Worker) 50*523fa7a6SAndroid Build Coastguard Worker 51*523fa7a6SAndroid Build Coastguard Workerpython_library( 52*523fa7a6SAndroid Build Coastguard Worker name = "arm_vela", 53*523fa7a6SAndroid Build Coastguard Worker srcs = [ 54*523fa7a6SAndroid Build Coastguard Worker "arm_vela.py", 55*523fa7a6SAndroid Build Coastguard Worker ], 56*523fa7a6SAndroid Build Coastguard Worker typing = True, 57*523fa7a6SAndroid Build Coastguard Worker deps = [ 58*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/ethos-u-vela:ethos-u-vela", 59*523fa7a6SAndroid Build Coastguard Worker ], 60*523fa7a6SAndroid Build Coastguard Worker) 61*523fa7a6SAndroid Build Coastguard Worker 62*523fa7a6SAndroid Build Coastguard Workerpython_library( 63*523fa7a6SAndroid Build Coastguard Worker name = "tosa_mapping", 64*523fa7a6SAndroid Build Coastguard Worker srcs = [ 65*523fa7a6SAndroid Build Coastguard Worker "tosa_mapping.py", 66*523fa7a6SAndroid Build Coastguard Worker ], 67*523fa7a6SAndroid Build Coastguard Worker typing = True, 68*523fa7a6SAndroid Build Coastguard Worker deps = [ 69*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/serializer:serializer", 70*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 71*523fa7a6SAndroid Build Coastguard Worker ], 72*523fa7a6SAndroid Build Coastguard Worker) 73*523fa7a6SAndroid Build Coastguard Worker 74*523fa7a6SAndroid Build Coastguard Workerpython_library( 75*523fa7a6SAndroid Build Coastguard Worker name = "tosa_quant_utils", 76*523fa7a6SAndroid Build Coastguard Worker srcs = [ 77*523fa7a6SAndroid Build Coastguard Worker "tosa_quant_utils.py", 78*523fa7a6SAndroid Build Coastguard Worker ], 79*523fa7a6SAndroid Build Coastguard Worker typing = True, 80*523fa7a6SAndroid Build Coastguard Worker deps = [ 81*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/numpy:numpy", 82*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/serializer:serializer", 83*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/tosa:tosa", 84*523fa7a6SAndroid Build Coastguard Worker ":tosa_mapping", 85*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 86*523fa7a6SAndroid Build Coastguard Worker ], 87*523fa7a6SAndroid Build Coastguard Worker) 88*523fa7a6SAndroid Build Coastguard Worker 89*523fa7a6SAndroid Build Coastguard Workerpython_library( 90*523fa7a6SAndroid Build Coastguard Worker name = "tosa_specification", 91*523fa7a6SAndroid Build Coastguard Worker srcs = [ 92*523fa7a6SAndroid Build Coastguard Worker "tosa_specification.py", 93*523fa7a6SAndroid Build Coastguard Worker ], 94*523fa7a6SAndroid Build Coastguard Worker typing = True, 95*523fa7a6SAndroid Build Coastguard Worker deps = [ 96*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/packaging:packaging", 97*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/backend:compile_spec_schema", 98*523fa7a6SAndroid Build Coastguard Worker ], 99*523fa7a6SAndroid Build Coastguard Worker) 100*523fa7a6SAndroid Build Coastguard Worker 101*523fa7a6SAndroid Build Coastguard Workerpython_library( 102*523fa7a6SAndroid Build Coastguard Worker name = "tosa_utils", 103*523fa7a6SAndroid Build Coastguard Worker srcs = [ 104*523fa7a6SAndroid Build Coastguard Worker "tosa_utils.py", 105*523fa7a6SAndroid Build Coastguard Worker ], 106*523fa7a6SAndroid Build Coastguard Worker typing = True, 107*523fa7a6SAndroid Build Coastguard Worker deps = [ 108*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/serialization_lib/python/serializer:serializer", 109*523fa7a6SAndroid Build Coastguard Worker ":tosa_quant_utils", 110*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/arm/operators:node_visitor", 111*523fa7a6SAndroid Build Coastguard Worker ], 112*523fa7a6SAndroid Build Coastguard Worker) 113