pytorch

1
# Description:
2
#   CUDA Templates for Linear Algebra Subroutines
3

4
load("@rules_cc//cc:defs.bzl", "cc_library")
5

6
cc_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