mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
f37a6523ef
Differential Revision: [D83685392](https://our.internmc.facebook.com/intern/diff/D83685392) Pull Request resolved: https://github.com/pytorch/pytorch/pull/164381 Approved by: https://github.com/janeyx99
21 lines
496 B
Python
21 lines
496 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__",
|
|
"//torch/headeronly:__pkg__",
|
|
],
|
|
)
|