minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

fix(lora): build the MoE LoRA align JIT kernel on ROCm (#36379)

Browse Source 
Co-authored-by: amdpilot-upstream-sync <amdpilot-upstream-sync@users.noreply.github.com>
This commit is contained in:
Yikai Zhang
2026-08-28 01:18:42 -07:00
committed by GitHub
co-authored by amdpilot-upstream-sync
parent 0c7d017dbb
commit d3b972cbf0
2 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/kernels/ops/moe/test_moe_lora_align_block_size.py
+2 -1
View File
@@ -9,9 +9,10 @@ import torch
# IMPORT PREBUILT KERNEL
# ---------------------------------------------------------
from sglang.kernels.ops.moe.moe_lora_align import moe_lora_align_block_size
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
register_cuda_ci(est_time=28, stage="base-b-kernel-unit", runner_config="1-gpu-large")
register_amd_ci(est_time=30, stage="jit-kernel-unit", runner_config="amd")
def round_up(x, base):