fix

fzyzcjy · fzyzcjy · commit 1583eb018d9b · 2025-09-19T16:54:03.000+08:00
diff --git a/csrc/fused_moe/cutlass_backend/cutlass_fused_moe_kernels.cuh b/csrc/fused_moe/cutlass_backend/cutlass_fused_moe_kernels.cuh
@@ -2482,7 +2482,7 @@ void doActivation(T* output, GemmOutputType const* gemm_result, float const* fp8
         return fn_list[static_cast<int>(activation_type.activation_type)];
       }
       if (num_experts_per_node == 64) {
-        constexpr int NUM_EXPERTS_PER_NODE_CONST = 128;
+        constexpr int NUM_EXPERTS_PER_NODE_CONST = 64;
         auto fn_list = std::array{
             &doActivationKernel<T, GemmOutputType, ScaleBiasType,
                                 IdentityAdaptor<cutlass::epilogue::thread::GELU>,

Original file line number	Diff line number	Diff line change
`@@ -2482,7 +2482,7 @@ void doActivation(T* output, GemmOutputType const* gemm_result, float const* fp8`
`2482`	`2482`	`return fn_list[static_cast<int>(activation_type.activation_type)];`
`2483`	`2483`	`}`
`2484`	`2484`	`if (num_experts_per_node == 64) {`
`2485`		`- constexpr int NUM_EXPERTS_PER_NODE_CONST = 128;`
	`2485`	`+ constexpr int NUM_EXPERTS_PER_NODE_CONST = 64;`
`2486`	`2486`	`auto fn_list = std::array{`
`2487`	`2487`	`&doActivationKernel<T, GemmOutputType, ScaleBiasType,`
`2488`	`2488`	`IdentityAdaptor<cutlass::epilogue::thread::GELU>,`