[Qwen] Remove cuda hard-code in qwen3 next (#25243)

wxsIcey · web-flow · commit 838d7116ba59 · 2025-09-19T12:25:12.000Z
Signed-off-by: Icey &lt;1790571317@qq.com&gt;
diff --git a/vllm/model_executor/models/qwen3_next.py b/vllm/model_executor/models/qwen3_next.py
@@ -306,7 +306,7 @@ def __init__(
             eps=self.layer_norm_epsilon,
             group_size=None,
             norm_before_gate=True,
-            device=torch.cuda.current_device(),
+            device=current_platform.current_device(),
             dtype=config.torch_dtype,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -306,7 +306,7 @@ def __init__(`
`306`	`306`	`eps=self.layer_norm_epsilon,`
`307`	`307`	`group_size=None,`
`308`	`308`	`norm_before_gate=True,`
`309`		`- device=torch.cuda.current_device(),`
	`309`	`+ device=current_platform.current_device(),`
`310`	`310`	`dtype=config.torch_dtype,`
`311`	`311`	`)`
`312`	`312`