[AMD] Register 3 JIT kernel unit tests for AMD CI (#27837)

Co-authored-by: michaelzhang-ai <michaelzhang@example.com>

test/registered/jit/test_clamp_position.py

@@ -4,10 +4,11 @@ import pytest
 import torch
 
 from sglang.jit_kernel.clamp_position import clamp_position_cuda
-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=12, suite="base-b-kernel-unit-1-gpu-large")
 register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
+register_amd_ci(est_time=12, suite="jit-kernel-unit-test-amd")
 
 
 def _reference_clamp_position(seq_lens):

test/registered/jit/test_resolve_future_token_ids.py

@@ -4,10 +4,11 @@ import pytest
 import torch
 
 from sglang.jit_kernel.resolve_future_token_ids import resolve_future_token_ids_cuda
-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=9, suite="base-b-kernel-unit-1-gpu-large")
 register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
+register_amd_ci(est_time=9, suite="jit-kernel-unit-test-amd")
 
 
 def _reference_resolve(input_ids, future_map):

test/registered/jit/test_rmsnorm_hf.py

@@ -11,10 +11,11 @@ from sglang.jit_kernel.rmsnorm_hf import (
     rmsnorm_hf,
 )
 from sglang.jit_kernel.utils import get_ci_test_range
-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=30, suite="base-b-kernel-unit-1-gpu-large")
 register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
+register_amd_ci(est_time=30, suite="jit-kernel-unit-test-amd")
 
 EPS = 1e-5
 DEVICE = "cuda"