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