build.bzl 491 B

1234567891011121314151617181920
  1. def define_targets(rules):
  2. rules.py_library(
  3. name = "autograd",
  4. srcs = rules.glob(["*.py"]),
  5. data = rules.glob([
  6. "*.yaml",
  7. "templates/*",
  8. ]),
  9. visibility = ["//:__subpackages__"],
  10. deps = [
  11. rules.requirement("PyYAML"),
  12. "//torchgen",
  13. ],
  14. )
  15. rules.filegroup(
  16. name = "deprecated_yaml",
  17. srcs = ["deprecated.yaml"],
  18. visibility = ["//:__subpackages__"],
  19. )