refine kernel selection

Yuening-wa · Yuening-wa · commit 269658b06dcd · 2025-08-06T00:48:58.000-07:00
Signed-off-by: Yuening Li &lt;62227368+yueningl@users.noreply.github.com&gt;
diff --git a/cpp/tensorrt_llm/thop/moeOp.cpp b/cpp/tensorrt_llm/thop/moeOp.cpp
@@ -108,13 +108,8 @@ class FusedMoeRunner : public torch::CustomClassHolder
                 return std::make_unique<
                     kernels::CutlassMoeFCRunner<__nv_fp8_e4m3, cutlass::uint4b_t, TypeAct, TypeAct>>();
             }
-            else
-            {
-                return std::make_unique<kernels::CutlassMoeFCRunner<TypeAct, cutlass::uint4b_t>>();
-            }
-#else
-            return std::make_unique<kernels::CutlassMoeFCRunner<TypeAct, cutlass::uint4b_t>>();
 #endif
+            return std::make_unique<kernels::CutlassMoeFCRunner<TypeAct, cutlass::uint4b_t>>();
         }
         else
         {

Original file line number	Diff line number	Diff line change
`@@ -108,13 +108,8 @@ class FusedMoeRunner : public torch::CustomClassHolder`
`108`	`108`	`return std::make_unique<`
`109`	`109`	`kernels::CutlassMoeFCRunner<__nv_fp8_e4m3, cutlass::uint4b_t, TypeAct, TypeAct>>();`
`110`	`110`	`}`
`111`		`- else`
`112`		`- {`
`113`		`- return std::make_unique<kernels::CutlassMoeFCRunner<TypeAct, cutlass::uint4b_t>>();`
`114`		`- }`
`115`		`-#else`
`116`		`- return std::make_unique<kernels::CutlassMoeFCRunner<TypeAct, cutlass::uint4b_t>>();`
`117`	`111`	`#endif`
	`112`	`+ return std::make_unique<kernels::CutlassMoeFCRunner<TypeAct, cutlass::uint4b_t>>();`
`118`	`113`	`}`
`119`	`114`	`else`
`120`	`115`	`{`