flagos-ai
diff --git a/‎vllm_fl/ops/fused_moe/fused_moe_utils.py‎
Lines changed: 1 addition & 1 deletion b/‎vllm_fl/ops/fused_moe/fused_moe_utils.py‎
Lines changed: 1 addition & 1 deletion
@@ -360,7 +360,7 @@ def apply(
                 )
                 return
 
-            elif current_platform.is_cuda():
+            elif current_platform.device_name == "nvidia":
                 # NVIDIA: let FlagGems own both expert GEMMs for unquantized and W8A16 inputs
                 import flag_gems
Original file line number	Diff line number	Diff line change
`@@ -360,7 +360,7 @@ def apply(`
`360`	`360`	`)`
`361`	`361`	`return`
`362`	`362`
`363`		`- elif current_platform.is_cuda():`
	`363`	`+ elif current_platform.device_name == "nvidia":`
`364`	`364`	`# NVIDIA: let FlagGems own both expert GEMMs for unquantized and W8A16 inputs`
`365`	`365`	`import flag_gems`
`366`	`366`