xref: /aosp_15_r20/external/pytorch/torchgen/BUCK.oss (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

python_library(
    name = "torchgen",
    srcs = glob(
        ["**/*.py"],
    ),
    base_module = "torchgen",
    visibility = ["PUBLIC"],
    deps = [
        "//third_party:pyyaml",
        "//third_party:typing-extensions",
    ],
)

python_binary(
    name = "gen",
    main_module = "torchgen.gen",
    visibility = [
        "PUBLIC",
    ],
    deps = [
        ":torchgen",
    ],
)