1*523fa7a6SAndroid Build Coastguard Worker# Copyright (c) Meta Platforms, Inc. and affiliates. 2*523fa7a6SAndroid Build Coastguard Worker# All rights reserved. 3*523fa7a6SAndroid Build Coastguard Worker# 4*523fa7a6SAndroid Build Coastguard Worker# This source code is licensed under the BSD-style license found in the 5*523fa7a6SAndroid Build Coastguard Worker# LICENSE file in the root directory of this source tree. 6*523fa7a6SAndroid Build Coastguard Worker 7*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:export_files.bzl", "export_file") 8*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:python_library.bzl", "python_library") 9*523fa7a6SAndroid Build Coastguard Workerload( 10*523fa7a6SAndroid Build Coastguard Worker "@fbsource//tools/build_defs:default_platform_defs.bzl", 11*523fa7a6SAndroid Build Coastguard Worker "CXX", 12*523fa7a6SAndroid Build Coastguard Worker) 13*523fa7a6SAndroid Build Coastguard Workerload("@fbsource//xplat/executorch/codegen:codegen.bzl", "executorch_generated_lib") 14*523fa7a6SAndroid Build Coastguard Workerload("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 15*523fa7a6SAndroid Build Coastguard Worker 16*523fa7a6SAndroid Build Coastguard Workeroncall("odai_jarvis") 17*523fa7a6SAndroid Build Coastguard Worker 18*523fa7a6SAndroid Build Coastguard Workerpython_library( 19*523fa7a6SAndroid Build Coastguard Worker name = "utils", 20*523fa7a6SAndroid Build Coastguard Worker srcs = [ 21*523fa7a6SAndroid Build Coastguard Worker "utils.py", 22*523fa7a6SAndroid Build Coastguard Worker ], 23*523fa7a6SAndroid Build Coastguard Worker deps = [ 24*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/tabulate:tabulate", 25*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 26*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 27*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:memory", 28*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 29*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects/edge:lib", 30*523fa7a6SAndroid Build Coastguard Worker ], 31*523fa7a6SAndroid Build Coastguard Worker) 32*523fa7a6SAndroid Build Coastguard Worker 33*523fa7a6SAndroid Build Coastguard Workerpython_library( 34*523fa7a6SAndroid Build Coastguard Worker name = "compiler", 35*523fa7a6SAndroid Build Coastguard Worker srcs = [ 36*523fa7a6SAndroid Build Coastguard Worker "compiler.py", 37*523fa7a6SAndroid Build Coastguard Worker ], 38*523fa7a6SAndroid Build Coastguard Worker deps = [ 39*523fa7a6SAndroid Build Coastguard Worker ":passes", 40*523fa7a6SAndroid Build Coastguard Worker ":utils", 41*523fa7a6SAndroid Build Coastguard Worker ":ops_registrations", 42*523fa7a6SAndroid Build Coastguard Worker ":replace_ops", 43*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 44*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot/quantizer:fusion_pass", 45*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot/quantizer:quantizer", 46*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/transforms:decompose_sdpa", 47*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/transforms:remove_clone_ops", 48*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:lib", 49*523fa7a6SAndroid Build Coastguard Worker "//executorch/devtools:lib", 50*523fa7a6SAndroid Build Coastguard Worker ], 51*523fa7a6SAndroid Build Coastguard Worker) 52*523fa7a6SAndroid Build Coastguard Worker 53*523fa7a6SAndroid Build Coastguard Worker 54*523fa7a6SAndroid Build Coastguard Workerpython_library( 55*523fa7a6SAndroid Build Coastguard Worker name = "pass_utils", 56*523fa7a6SAndroid Build Coastguard Worker srcs = [ 57*523fa7a6SAndroid Build Coastguard Worker "pass_utils.py", 58*523fa7a6SAndroid Build Coastguard Worker ], 59*523fa7a6SAndroid Build Coastguard Worker deps = [ 60*523fa7a6SAndroid Build Coastguard Worker ":utils", 61*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 62*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 63*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 64*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 65*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:spec_prop_pass", 66*523fa7a6SAndroid Build Coastguard Worker ], 67*523fa7a6SAndroid Build Coastguard Worker) 68*523fa7a6SAndroid Build Coastguard Worker 69*523fa7a6SAndroid Build Coastguard Workerpython_library( 70*523fa7a6SAndroid Build Coastguard Worker name = "passes", 71*523fa7a6SAndroid Build Coastguard Worker srcs = [ 72*523fa7a6SAndroid Build Coastguard Worker "_passes.py", 73*523fa7a6SAndroid Build Coastguard Worker ], 74*523fa7a6SAndroid Build Coastguard Worker deps = [ 75*523fa7a6SAndroid Build Coastguard Worker ":utils", 76*523fa7a6SAndroid Build Coastguard Worker ":fuse_ops", 77*523fa7a6SAndroid Build Coastguard Worker ":simplify_ops", 78*523fa7a6SAndroid Build Coastguard Worker ":replace_ops", 79*523fa7a6SAndroid Build Coastguard Worker ":reorder_ops", 80*523fa7a6SAndroid Build Coastguard Worker ":remove_ops", 81*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 82*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 83*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 84*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 85*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:spec_prop_pass", 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 = "ops_registrations", 91*523fa7a6SAndroid Build Coastguard Worker srcs = [ 92*523fa7a6SAndroid Build Coastguard Worker "ops_registrations.py", 93*523fa7a6SAndroid Build Coastguard Worker ], 94*523fa7a6SAndroid Build Coastguard Worker deps = [ 95*523fa7a6SAndroid Build Coastguard Worker "fbcode//caffe2:torch", 96*523fa7a6SAndroid Build Coastguard Worker "fbcode//executorch/exir:scalar_type", 97*523fa7a6SAndroid Build Coastguard Worker "fbcode//executorch/backends/cadence/aot:utils", 98*523fa7a6SAndroid Build Coastguard Worker ], 99*523fa7a6SAndroid Build Coastguard Worker) 100*523fa7a6SAndroid Build Coastguard Worker 101*523fa7a6SAndroid Build Coastguard Workerexport_file(name = "functions.yaml") 102*523fa7a6SAndroid Build Coastguard Worker 103*523fa7a6SAndroid Build Coastguard Workerexecutorch_generated_lib( 104*523fa7a6SAndroid Build Coastguard Worker name = "cadence_aot_lib", 105*523fa7a6SAndroid Build Coastguard Worker custom_ops_yaml_target = "//executorch/kernels/portable:custom_ops.yaml", 106*523fa7a6SAndroid Build Coastguard Worker functions_yaml_target = ":functions.yaml", 107*523fa7a6SAndroid Build Coastguard Worker platforms = CXX, 108*523fa7a6SAndroid Build Coastguard Worker visibility = ["PUBLIC"], 109*523fa7a6SAndroid Build Coastguard Worker deps = [ 110*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/reference/kernels:cadence_kernels", 111*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/reference/operators:cadence_cpu_ops", 112*523fa7a6SAndroid Build Coastguard Worker "//executorch/kernels/portable:executorch_all_ops", 113*523fa7a6SAndroid Build Coastguard Worker "//executorch/kernels/portable:operators", 114*523fa7a6SAndroid Build Coastguard Worker ], 115*523fa7a6SAndroid Build Coastguard Worker) 116*523fa7a6SAndroid Build Coastguard Worker 117*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 118*523fa7a6SAndroid Build Coastguard Worker name = "test_pass_filter", 119*523fa7a6SAndroid Build Coastguard Worker srcs = [ 120*523fa7a6SAndroid Build Coastguard Worker "tests/test_pass_filter.py", 121*523fa7a6SAndroid Build Coastguard Worker ], 122*523fa7a6SAndroid Build Coastguard Worker typing = True, 123*523fa7a6SAndroid Build Coastguard Worker deps = [ 124*523fa7a6SAndroid Build Coastguard Worker ":pass_utils", 125*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 126*523fa7a6SAndroid Build Coastguard Worker ], 127*523fa7a6SAndroid Build Coastguard Worker) 128*523fa7a6SAndroid Build Coastguard Worker 129*523fa7a6SAndroid Build Coastguard Workerpython_library( 130*523fa7a6SAndroid Build Coastguard Worker name = "compiler_utils", 131*523fa7a6SAndroid Build Coastguard Worker srcs = [ 132*523fa7a6SAndroid Build Coastguard Worker "compiler_utils.py", 133*523fa7a6SAndroid Build Coastguard Worker ], 134*523fa7a6SAndroid Build Coastguard Worker typing = True, 135*523fa7a6SAndroid Build Coastguard Worker deps = [ 136*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 137*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 138*523fa7a6SAndroid Build Coastguard Worker ], 139*523fa7a6SAndroid Build Coastguard Worker) 140*523fa7a6SAndroid Build Coastguard Worker 141*523fa7a6SAndroid Build Coastguard Workerpython_library( 142*523fa7a6SAndroid Build Coastguard Worker name = "graph_builder", 143*523fa7a6SAndroid Build Coastguard Worker srcs = [ 144*523fa7a6SAndroid Build Coastguard Worker "graph_builder.py", 145*523fa7a6SAndroid Build Coastguard Worker ], 146*523fa7a6SAndroid Build Coastguard Worker typing = True, 147*523fa7a6SAndroid Build Coastguard Worker deps = [ 148*523fa7a6SAndroid Build Coastguard Worker "fbcode//caffe2:torch", 149*523fa7a6SAndroid Build Coastguard Worker "fbcode//executorch/exir:pass_base", 150*523fa7a6SAndroid Build Coastguard Worker ], 151*523fa7a6SAndroid Build Coastguard Worker) 152*523fa7a6SAndroid Build Coastguard Worker 153*523fa7a6SAndroid Build Coastguard Workerpython_library( 154*523fa7a6SAndroid Build Coastguard Worker name = "fuse_ops", 155*523fa7a6SAndroid Build Coastguard Worker srcs = [ 156*523fa7a6SAndroid Build Coastguard Worker "fuse_ops.py", 157*523fa7a6SAndroid Build Coastguard Worker ], 158*523fa7a6SAndroid Build Coastguard Worker typing = True, 159*523fa7a6SAndroid Build Coastguard Worker deps = [ 160*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 161*523fa7a6SAndroid Build Coastguard Worker ":compiler_utils", 162*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 163*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:utils", 164*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 165*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 166*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects/edge:lib", 167*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 168*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:spec_prop_pass", 169*523fa7a6SAndroid Build Coastguard Worker ], 170*523fa7a6SAndroid Build Coastguard Worker) 171*523fa7a6SAndroid Build Coastguard Worker 172*523fa7a6SAndroid Build Coastguard Workerpython_library( 173*523fa7a6SAndroid Build Coastguard Worker name = "simplify_ops", 174*523fa7a6SAndroid Build Coastguard Worker srcs = [ 175*523fa7a6SAndroid Build Coastguard Worker "simplify_ops.py", 176*523fa7a6SAndroid Build Coastguard Worker ], 177*523fa7a6SAndroid Build Coastguard Worker typing = True, 178*523fa7a6SAndroid Build Coastguard Worker deps = [ 179*523fa7a6SAndroid Build Coastguard Worker ":pass_utils", 180*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 181*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 182*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 183*523fa7a6SAndroid Build Coastguard Worker ], 184*523fa7a6SAndroid Build Coastguard Worker) 185*523fa7a6SAndroid Build Coastguard Worker 186*523fa7a6SAndroid Build Coastguard Workerpython_library( 187*523fa7a6SAndroid Build Coastguard Worker name = "remove_ops", 188*523fa7a6SAndroid Build Coastguard Worker srcs = [ 189*523fa7a6SAndroid Build Coastguard Worker "remove_ops.py", 190*523fa7a6SAndroid Build Coastguard Worker ], 191*523fa7a6SAndroid Build Coastguard Worker typing = True, 192*523fa7a6SAndroid Build Coastguard Worker deps = [ 193*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 194*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 195*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:simplify_ops", 196*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 197*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 198*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects/edge:lib", 199*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:spec_prop_pass", 200*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/transforms:remove_clone_ops" 201*523fa7a6SAndroid Build Coastguard Worker ], 202*523fa7a6SAndroid Build Coastguard Worker) 203*523fa7a6SAndroid Build Coastguard Worker 204*523fa7a6SAndroid Build Coastguard Workerpython_library( 205*523fa7a6SAndroid Build Coastguard Worker name = "reorder_ops", 206*523fa7a6SAndroid Build Coastguard Worker srcs = [ 207*523fa7a6SAndroid Build Coastguard Worker "reorder_ops.py", 208*523fa7a6SAndroid Build Coastguard Worker ], 209*523fa7a6SAndroid Build Coastguard Worker typing = True, 210*523fa7a6SAndroid Build Coastguard Worker deps = [ 211*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 212*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler_utils", 213*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 214*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:utils", 215*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 216*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:tensor", 217*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 218*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects/edge:lib", 219*523fa7a6SAndroid Build Coastguard Worker ], 220*523fa7a6SAndroid Build Coastguard Worker) 221*523fa7a6SAndroid Build Coastguard Worker 222*523fa7a6SAndroid Build Coastguard Workerpython_library( 223*523fa7a6SAndroid Build Coastguard Worker name = "replace_ops", 224*523fa7a6SAndroid Build Coastguard Worker srcs = [ 225*523fa7a6SAndroid Build Coastguard Worker "replace_ops.py", 226*523fa7a6SAndroid Build Coastguard Worker ], 227*523fa7a6SAndroid Build Coastguard Worker typing = True, 228*523fa7a6SAndroid Build Coastguard Worker deps = [ 229*523fa7a6SAndroid Build Coastguard Worker ":pass_utils", 230*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 231*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler_utils", 232*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:fuse_ops", 233*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 234*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:remove_ops", 235*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:utils", 236*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 237*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 238*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects/edge:lib", 239*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:spec_prop_pass", 240*523fa7a6SAndroid Build Coastguard Worker ], 241*523fa7a6SAndroid Build Coastguard Worker) 242*523fa7a6SAndroid Build Coastguard Worker 243*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 244*523fa7a6SAndroid Build Coastguard Worker name = "test_graph_builder", 245*523fa7a6SAndroid Build Coastguard Worker srcs = [ 246*523fa7a6SAndroid Build Coastguard Worker "tests/test_graph_builder.py", 247*523fa7a6SAndroid Build Coastguard Worker ], 248*523fa7a6SAndroid Build Coastguard Worker typing = True, 249*523fa7a6SAndroid Build Coastguard Worker deps = [ 250*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 251*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:graph_builder", 252*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 253*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 254*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 255*523fa7a6SAndroid Build Coastguard Worker "//later:lib", 256*523fa7a6SAndroid Build Coastguard Worker ":ops_registrations" 257*523fa7a6SAndroid Build Coastguard Worker ], 258*523fa7a6SAndroid Build Coastguard Worker) 259*523fa7a6SAndroid Build Coastguard Worker 260*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 261*523fa7a6SAndroid Build Coastguard Worker name = "test_replace_ops_passes", 262*523fa7a6SAndroid Build Coastguard Worker srcs = [ 263*523fa7a6SAndroid Build Coastguard Worker "tests/test_replace_ops_passes.py", 264*523fa7a6SAndroid Build Coastguard Worker ], 265*523fa7a6SAndroid Build Coastguard Worker supports_static_listing = False, 266*523fa7a6SAndroid Build Coastguard Worker typing = True, 267*523fa7a6SAndroid Build Coastguard Worker deps = [ 268*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/parameterized:parameterized", 269*523fa7a6SAndroid Build Coastguard Worker ":compiler", 270*523fa7a6SAndroid Build Coastguard Worker ":replace_ops", 271*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 272*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler", 273*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:graph_builder", 274*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 275*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir:pass_base", 276*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 277*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/passes:lib", 278*523fa7a6SAndroid Build Coastguard Worker ], 279*523fa7a6SAndroid Build Coastguard Worker) 280*523fa7a6SAndroid Build Coastguard Worker 281*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 282*523fa7a6SAndroid Build Coastguard Worker name = "test_fusion_ops_passes", 283*523fa7a6SAndroid Build Coastguard Worker srcs = [ 284*523fa7a6SAndroid Build Coastguard Worker "tests/test_fusion_ops_passes.py", 285*523fa7a6SAndroid Build Coastguard Worker ], 286*523fa7a6SAndroid Build Coastguard Worker typing = True, 287*523fa7a6SAndroid Build Coastguard Worker deps = [ 288*523fa7a6SAndroid Build Coastguard Worker ":compiler", 289*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 290*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler", 291*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:fuse_ops", 292*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:graph_builder", 293*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:ops_registrations", 294*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 295*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 296*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects/edge:lib", 297*523fa7a6SAndroid Build Coastguard Worker ], 298*523fa7a6SAndroid Build Coastguard Worker) 299*523fa7a6SAndroid Build Coastguard Worker 300*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 301*523fa7a6SAndroid Build Coastguard Worker name = "test_remove_ops_passes", 302*523fa7a6SAndroid Build Coastguard Worker srcs = [ 303*523fa7a6SAndroid Build Coastguard Worker "tests/test_remove_ops_passes.py", 304*523fa7a6SAndroid Build Coastguard Worker ], 305*523fa7a6SAndroid Build Coastguard Worker supports_static_listing = False, 306*523fa7a6SAndroid Build Coastguard Worker typing = True, 307*523fa7a6SAndroid Build Coastguard Worker deps = [ 308*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/parameterized:parameterized", 309*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/pyre-extensions:pyre-extensions", 310*523fa7a6SAndroid Build Coastguard Worker ":compiler", 311*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 312*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler", 313*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:ops_registrations", 314*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 315*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:remove_ops", 316*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot/quantizer:quantizer", 317*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 318*523fa7a6SAndroid Build Coastguard Worker ], 319*523fa7a6SAndroid Build Coastguard Worker) 320*523fa7a6SAndroid Build Coastguard Worker 321*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 322*523fa7a6SAndroid Build Coastguard Worker name = "test_simplify_ops_passes", 323*523fa7a6SAndroid Build Coastguard Worker srcs = [ 324*523fa7a6SAndroid Build Coastguard Worker "tests/test_simplify_ops_passes.py", 325*523fa7a6SAndroid Build Coastguard Worker ], 326*523fa7a6SAndroid Build Coastguard Worker supports_static_listing = False, 327*523fa7a6SAndroid Build Coastguard Worker typing = True, 328*523fa7a6SAndroid Build Coastguard Worker deps = [ 329*523fa7a6SAndroid Build Coastguard Worker "fbsource//third-party/pypi/parameterized:parameterized", 330*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 331*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler", 332*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:ops_registrations", 333*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 334*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:simplify_ops", 335*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 336*523fa7a6SAndroid Build Coastguard Worker ], 337*523fa7a6SAndroid Build Coastguard Worker) 338*523fa7a6SAndroid Build Coastguard Worker 339*523fa7a6SAndroid Build Coastguard Workerpython_unittest( 340*523fa7a6SAndroid Build Coastguard Worker name = "test_reorder_ops_passes", 341*523fa7a6SAndroid Build Coastguard Worker srcs = [ 342*523fa7a6SAndroid Build Coastguard Worker "tests/test_reorder_ops_passes.py", 343*523fa7a6SAndroid Build Coastguard Worker ], 344*523fa7a6SAndroid Build Coastguard Worker typing = True, 345*523fa7a6SAndroid Build Coastguard Worker deps = [ 346*523fa7a6SAndroid Build Coastguard Worker ":compiler", 347*523fa7a6SAndroid Build Coastguard Worker ":pass_utils", 348*523fa7a6SAndroid Build Coastguard Worker "//caffe2:torch", 349*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:compiler", 350*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:fuse_ops", 351*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:ops_registrations", 352*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:pass_utils", 353*523fa7a6SAndroid Build Coastguard Worker "//executorch/backends/cadence/aot:reorder_ops", 354*523fa7a6SAndroid Build Coastguard Worker "//executorch/exir/dialects:lib", 355*523fa7a6SAndroid Build Coastguard Worker ], 356*523fa7a6SAndroid Build Coastguard Worker) 357