EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

AMD/ROCm: update AITER repo to ROCm/aiter (#3747)

Browse Source
This commit is contained in:
HAI
2025-02-21 00:18:08 -08:00
committed by GitHub
parent c6a4852136
commit 5c54ef0352
3 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/sglang/srt/layers/moe/fused_moe_triton/layer.py
View File

@@ -167,8 +167,8 @@ class UnquantizedFusedMoEMethod(FusedMoEMethodBase, CustomOp):
)
if is_hip_ and get_bool_env_var("CK_MOE"):
import ater
from ater.fused_moe import fused_experts_ck
import aiter
from aiter.fused_moe import fused_experts_ck
assert activation == "silu", f"{activation=} is not supported."

4
python/sglang/srt/layers/quantization/fp8.py
View File

@@ -789,8 +789,8 @@ class Fp8MoEMethod:
)
if is_hip_ and get_bool_env_var("CK_MOE"):
import ater
from ater.fused_moe import fused_experts_ck
import aiter
from aiter.fused_moe import fused_experts_ck
assert activation == "silu", f"{activation=} is not supported."