STT-tensorflow/third_party/cub.BUILD

28 lines
552 B
Plaintext

# Description: CUB library which is a set of primitives for GPU programming.
package(
default_visibility = ["//visibility:public"],
)
licenses(["notice"]) # BSD
exports_files(["LICENSE.TXT"])
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts", "if_cuda")
filegroup(
name = "cub_header_files",
srcs = glob([
"cub/**",
]),
)
cc_library(
name = "cub",
hdrs = if_cuda([":cub_header_files"]),
include_prefix = "third_party",
deps = [
"@local_config_cuda//cuda:cuda_headers",
],
)