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