third_party/cub.BUILD

# 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",
    ],
)