diff --git a/test/registered/kernels/test_gather_spec_extras.py b/test/registered/kernels/test_gather_spec_extras.py index 0d71ae7a2..a2ad11fb3 100644 --- a/test/registered/kernels/test_gather_spec_extras.py +++ b/test/registered/kernels/test_gather_spec_extras.py @@ -1,6 +1,7 @@ -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=10, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=10, suite="nightly-amd-kernel-1-gpu", nightly=True) import unittest