add tma in 90 rear

Author: da.huo

src/turbomind/kernels/gemm/CMakeLists.txt

@@ -16,7 +16,6 @@ set(GEMM2_KERNELS_SM80
     kernel/sm80_16816_16.cu
 )
 set(GEMM2_KERNELS_SM90
-    tma.cu
     kernel/sm90_16816_4.cu
     kernel/sm90_16816_8.cu
     kernel/sm90_16816_16.cu
@@ -87,7 +86,7 @@ set_property(TARGET gemm2 PROPERTY CUDA_RESOLVE_DEVICE_SYMBOLS ON)
 
 if(GEMM2_ARCH_90_ENABLED)
     # SM90 kernels only compile for 90/90a; avoid building them for sm_100.
-    add_library(gemm2_sm90 STATIC ${GEMM2_KERNELS_SM90})
+    add_library(gemm2_sm90 STATIC ${GEMM2_KERNELS_SM90} tma.cu )
     set_target_properties(gemm2_sm90 PROPERTIES
         CUDA_ARCHITECTURES "${_sm90_archs}"
         POSITION_INDEPENDENT_CODE ON