pytorch

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