pytorch

1
def 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