18 lines
431 B
Python
18 lines
431 B
Python
def define_targets(rules):
|
|
rules.py_binary(
|
|
name = "generate_code",
|
|
srcs = ["generate_code.py"],
|
|
visibility = ["//:__pkg__"],
|
|
deps = [
|
|
rules.requirement("PyYAML"),
|
|
"//tools/autograd",
|
|
"//torchgen",
|
|
],
|
|
)
|
|
|
|
rules.py_binary(
|
|
name = "gen_version_header",
|
|
srcs = ["gen_version_header.py"],
|
|
visibility = ["//:__pkg__"],
|
|
)
|