Fix template instantiation

tlc-pack · Dec 14, 2023 · d489188 · d489188
1 parent 7191fbd
commit d489188
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_fp16.cu b/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_fp16.cu
@@ -19,7 +19,7 @@
 namespace fastertransformer
 {
 
-template <>
+template
 void moe_gemm_bias_act(const half* A, const half* B, const half* weight_scales, const half* biases, half* C,
     int64_t* total_rows_before_expert, int64_t total_rows, int64_t gemm_n, int64_t gemm_k, int num_experts,
     std::optional<std::string> activation, cudaStream_t stream);

diff --git a/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_uint4.cu b/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_uint4.cu
@@ -19,7 +19,7 @@
 namespace fastertransformer
 {
 
-template <>
+template
 void moe_gemm_bias_act(const half* A, const cutlass::uint4b_t* B, const half* weight_scales, const half* biases,
     half* C, int64_t* total_rows_before_expert, int64_t total_rows, int64_t gemm_n, int64_t gemm_k, int num_experts,
     std::optional<std::string> activation, cudaStream_t stream);

diff --git a/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_uint8.cu b/cutlass_kernels/moe_gemm/moe_gemm_kernels_fp16_uint8.cu
@@ -19,9 +19,9 @@
 namespace fastertransformer
 {
 
-template <>
+template
 void moe_gemm_bias_act(const half* A, const uint8_t* B, const half* weight_scales, const half* biases, half* C,
     int64_t* total_rows_before_expert, int64_t total_rows, int64_t gemm_n, int64_t gemm_k, int num_experts,
     std::optional<std::string> activation, cudaStream_t stream);
 
-}
+}