diff --git a/test/registered/jit/test_fused_metadata_copy.py b/test/registered/jit/test_fused_metadata_copy.py index f2229f197..82b9ff6f7 100644 --- a/test/registered/jit/test_fused_metadata_copy.py +++ b/test/registered/jit/test_fused_metadata_copy.py @@ -14,10 +14,11 @@ import time import pytest import torch -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=100, stage="base-b-kernel-unit", runner_config="1-gpu-large") register_cuda_ci(est_time=400, suite="nightly-kernel-1-gpu", nightly=True) +register_amd_ci(est_time=100, suite="nightly-amd-kernel-1-gpu", nightly=True) # ============================================================================= # Helper Functions