[refactor] remove redundant code in linear

2025-12-24 12:02:09 +08:00
parent 75d0bdae2f
commit 383eb5459a
8 changed files with 37 additions and 1570 deletions
--- a/vllm_kunlun/models/qwen3_moe.py
+++ b/vllm_kunlun/models/qwen3_moe.py
@@ -40,7 +40,7 @@ from vllm.logger import init_logger
 from vllm_kunlun.ops.activation import SiluAndMul
 from vllm_kunlun.ops.fused_moe.layer import FusedMoE
 from vllm.model_executor.layers.layernorm import RMSNorm
-from vllm_kunlun.ops.linear import (MergedColumnParallelLinear,
+from vllm.model_executor.layers.linear import (MergedColumnParallelLinear,
                                               QKVParallelLinear,
                                               RowParallelLinear,
                                               ReplicatedLinear)