1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
|
def define_targets(rules):
rules.py_library(
name = "torchgen",
srcs = rules.glob(["**/*.py"]),
visibility = ["//visibility:public"],
deps = [
rules.requirement("PyYAML"),
rules.requirement("typing-extensions"),
],
)
rules.py_binary(
name = "gen",
srcs = [":torchgen"],
visibility = ["//visibility:public"],
deps = [
rules.requirement("PyYAML"),
rules.requirement("typing-extensions"),
],
)
rules.py_binary(
name = "gen_executorch",
srcs = [":torchgen"],
visibility = ["//visibility:public"],
deps = [
rules.requirement("PyYAML"),
rules.requirement("typing-extensions"),
],
)
|