1load("@fbcode_macros//build_defs:python_binary.bzl", "python_binary") 2load("@fbcode_macros//build_defs:python_library.bzl", "python_library") 3 4oncall("ai_infra_mobile_platform") 5 6python_library( 7 name = "lib", 8 srcs = [ 9 "gen.py", 10 "utils.py", 11 ], 12 deps = [ 13 "fbsource//third-party/pypi/ruamel-yaml:ruamel-yaml", 14 "//caffe2:torch", 15 "//executorch/exir/dialects/edge/arg:lib", 16 "//executorch/exir/dialects/edge/dtype:lib", 17 "//executorch/exir/dialects/edge/op:lib", 18 ], 19) 20 21python_binary( 22 name = "gen", 23 srcs = [], 24 main_function = "executorch.exir.dialects.edge.spec.gen.main", 25 deps = [ 26 "fbsource//third-party/pypi/expecttest:expecttest", # @manual 27 ":lib", 28 ], 29) 30