mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71413 ghstack-source-id: 150235101 Test Plan: Verified manually. Rely on CI to validate. Reviewed By: malfet Differential Revision: D33635740 fbshipit-source-id: 82c6798a20c01c16fb17547d4a0ba30e6ffc272d (cherry picked from commit d7a0b39f510f59fe16f138a712d380e0091b230a)
20 lines
734 B
Python
20 lines
734 B
Python
load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_library", "cc_test")
|
|
load("@rules_cuda//cuda:defs.bzl", "requires_cuda_enabled")
|
|
load("//c10/macros:cmake_configure_file.bzl", "cmake_configure_file")
|
|
load("//tools/config:defs.bzl", "if_cuda")
|
|
|
|
# Rules implementation for the Bazel build system. Since the common
|
|
# build structure aims to replicate Bazel as much as possible, most of
|
|
# the rules simply forward to the Bazel definitions.
|
|
rules = struct(
|
|
cc_binary = cc_binary,
|
|
cc_library = cc_library,
|
|
cc_test = cc_test,
|
|
cmake_configure_file = cmake_configure_file,
|
|
filegroup = native.filegroup,
|
|
glob = native.glob,
|
|
if_cuda = if_cuda,
|
|
requires_cuda_enabled = requires_cuda_enabled,
|
|
select = select,
|
|
)
|