mirror of
https://github.com/tensorflow/tensorflow.git
synced 2024-11-21 21:05:19 +00:00
38 lines
744 B
Plaintext
38 lines
744 B
Plaintext
# Description:
|
|
# CUTLASS is a collection of CUDA C++ template abstractions for implementing high-performance
|
|
# matrix-matrix multiplication (GEMM) and related computations at all levels and scales within CUDA.
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
)
|
|
|
|
licenses(["notice"]) # MIT
|
|
|
|
exports_files(["LICENSE.txt"])
|
|
|
|
filegroup(
|
|
name = "cutlass_header_files",
|
|
srcs = glob([
|
|
"include/**",
|
|
]),
|
|
)
|
|
|
|
filegroup(
|
|
name = "cutlass_util_header_files",
|
|
srcs = glob([
|
|
"tools/util/include/**",
|
|
]),
|
|
)
|
|
|
|
cc_library(
|
|
name = "cutlass",
|
|
hdrs = [
|
|
":cutlass_header_files",
|
|
":cutlass_util_header_files",
|
|
],
|
|
includes = [
|
|
"include",
|
|
"tools/util/include",
|
|
],
|
|
)
|