xref: /aosp_15_r20/external/pytorch/tools/setup_helpers/build.bzl (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1def define_targets(rules):
2    rules.py_binary(
3        name = "generate_code",
4        srcs = ["generate_code.py"],
5        visibility = ["//:__pkg__"],
6        deps = [
7            rules.requirement("PyYAML"),
8            "//tools/autograd",
9            "//torchgen",
10        ],
11    )
12
13    rules.py_binary(
14        name = "gen_version_header",
15        srcs = ["gen_version_header.py"],
16        visibility = ["//:__pkg__"],
17    )
18