Skip to content

Commit 6927384

Browse files
author
da.huo
committed
add other tma
1 parent 4345aa5 commit 6927384

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/turbomind/kernels/gemm/CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ add_library(gemm2
5050
cast.cu
5151
unpack.cu
5252
context.cu
53+
tma.cu
5354
tuner/cache_utils.cu
5455
tuner/measurer.cu
5556
tuner/sampler.cu
@@ -86,7 +87,7 @@ set_property(TARGET gemm2 PROPERTY CUDA_RESOLVE_DEVICE_SYMBOLS ON)
8687

8788
if(GEMM2_ARCH_90_ENABLED)
8889
# SM90 kernels only compile for 90/90a; avoid building them for sm_100.
89-
add_library(gemm2_sm90 STATIC ${GEMM2_KERNELS_SM90})
90+
add_library(gemm2_sm90 STATIC ${GEMM2_KERNELS_SM90} )
9091
set_target_properties(gemm2_sm90 PROPERTIES
9192
CUDA_ARCHITECTURES "${_sm90_archs}"
9293
POSITION_INDEPENDENT_CODE ON

0 commit comments

Comments
 (0)