[AMD] Register 2 CPU/triton unit + kernel tests for AMD 1-GPU PR CI (#31379)

This commit is contained in:
Michael
2026-07-17 09:01:46 -07:00
committed by GitHub
parent 5e7eed4c00
commit c546afc147
2 changed files with 4 additions and 2 deletions
+2 -1
View File
@@ -3,10 +3,11 @@ import unittest
import torch
from sglang.kernels.ops.moe.ep_moe_kernels import zero_experts_compute_triton
from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=5, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=5, stage="stage-b", runner_config="1-gpu-small-amd")
@unittest.skipIf(not torch.cuda.is_available(), "CUDA is required")
@@ -2,9 +2,10 @@ import io
import pytest
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=2, stage="base-b", runner_config="1-gpu-small")
register_amd_ci(est_time=2, stage="stage-b", runner_config="1-gpu-small-amd")
from sglang.srt.distributed.device_communicators import cuda_wrapper