27 lines
657 B
Plaintext
27 lines
657 B
Plaintext
# Description:
|
|
# CUDA Templates for Linear Algebra Subroutines
|
|
|
|
load("@rules_cc//cc:defs.bzl", "cc_library")
|
|
|
|
cc_library(
|
|
name = "cutlass",
|
|
hdrs = glob([
|
|
"include/**/*.h",
|
|
"include/**/*.hpp",
|
|
"include/**/*.inl",
|
|
"tools/util/include/**/*.h",
|
|
"tools/util/include/**/*.hpp",
|
|
"tools/util/include/**/*.inl",
|
|
]),
|
|
defines = [
|
|
"CUTLASS_ENABLE_TENSOR_CORE_MMA=1",
|
|
"CUTLASS_ENABLE_SM90_EXTENDED_MMA_SHAPES=1",
|
|
"CUTE_SM90_EXTENDED_MMA_SHAPES_ENABLED",
|
|
],
|
|
includes = [
|
|
"include/",
|
|
"tools/util/include/",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|