xref: /aosp_15_r20/external/pytorch/third_party/cutlass.BUILD (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1*da0073e9SAndroid Build Coastguard Worker# Description:
2*da0073e9SAndroid Build Coastguard Worker#   CUDA Templates for Linear Algebra Subroutines
3*da0073e9SAndroid Build Coastguard Worker
4*da0073e9SAndroid Build Coastguard Workerload("@rules_cc//cc:defs.bzl", "cc_library")
5*da0073e9SAndroid Build Coastguard Worker
6*da0073e9SAndroid Build Coastguard Workercc_library(
7*da0073e9SAndroid Build Coastguard Worker    name = "cutlass",
8*da0073e9SAndroid Build Coastguard Worker    hdrs = glob([
9*da0073e9SAndroid Build Coastguard Worker        "include/**/*.h",
10*da0073e9SAndroid Build Coastguard Worker        "include/**/*.hpp",
11*da0073e9SAndroid Build Coastguard Worker        "include/**/*.inl",
12*da0073e9SAndroid Build Coastguard Worker        "tools/util/include/**/*.h",
13*da0073e9SAndroid Build Coastguard Worker        "tools/util/include/**/*.hpp",
14*da0073e9SAndroid Build Coastguard Worker        "tools/util/include/**/*.inl",
15*da0073e9SAndroid Build Coastguard Worker    ]),
16*da0073e9SAndroid Build Coastguard Worker    includes = [
17*da0073e9SAndroid Build Coastguard Worker        "include/",
18*da0073e9SAndroid Build Coastguard Worker        "tools/util/include/",
19*da0073e9SAndroid Build Coastguard Worker    ],
20*da0073e9SAndroid Build Coastguard Worker    visibility = ["//visibility:public"],
21*da0073e9SAndroid Build Coastguard Worker)
22