xref: /aosp_15_r20/external/tensorflow/third_party/cub.BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1*b6fb3261SAndroid Build Coastguard Worker# Description: CUB library which is a set of primitives for GPU programming.
2*b6fb3261SAndroid Build Coastguard Worker
3*b6fb3261SAndroid Build Coastguard Workerpackage(
4*b6fb3261SAndroid Build Coastguard Worker    default_visibility = ["//visibility:public"],
5*b6fb3261SAndroid Build Coastguard Worker)
6*b6fb3261SAndroid Build Coastguard Worker
7*b6fb3261SAndroid Build Coastguard Workerlicenses(["notice"])  # BSD
8*b6fb3261SAndroid Build Coastguard Worker
9*b6fb3261SAndroid Build Coastguard Workerexports_files(["LICENSE.TXT"])
10*b6fb3261SAndroid Build Coastguard Worker
11*b6fb3261SAndroid Build Coastguard Workercc_library(
12*b6fb3261SAndroid Build Coastguard Worker    name = "cub",
13*b6fb3261SAndroid Build Coastguard Worker    hdrs = glob(["cub/**"]),
14*b6fb3261SAndroid Build Coastguard Worker    deps = ["@local_config_cuda//cuda:cuda_headers"],
15*b6fb3261SAndroid Build Coastguard Worker)
16