mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
59e261bbd8
This reverts commit f5130bf339f12ccf5c6296130c47685bdc4858e4. Reverted https://github.com/pytorch/pytorch/pull/158720 on behalf of https://github.com/yangw-dev due to this pr failed internally when build torchgen due to rror: fail: Unknown PyPI project: pyyaml, it seems like this is caused by change PyYAML into pyyaml, please fix it ([comment](https://github.com/pytorch/pytorch/pull/158720#issuecomment-3129995414))
21 lines
536 B
Python
21 lines
536 B
Python
def define_targets(rules):
|
|
rules.py_library(
|
|
name = "torchgen",
|
|
srcs = rules.glob(["**/*.py"]),
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
rules.requirement("PyYAML"),
|
|
rules.requirement("typing-extensions"),
|
|
],
|
|
)
|
|
|
|
rules.py_binary(
|
|
name = "gen",
|
|
srcs = [":torchgen"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
rules.requirement("PyYAML"),
|
|
rules.requirement("typing-extensions"),
|
|
],
|
|
)
|