diff --git a/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py b/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py
index f56c8df2a..5551d2968 100644
--- a/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py
+++ b/python/sglang/jit_kernel/benchmark/bench_resolve_future_token_ids.py
@@ -11,9 +11,10 @@ from sglang.jit_kernel.benchmark.utils import (
 )
 from sglang.jit_kernel.resolve_future_token_ids import resolve_future_token_ids_cuda
 from sglang.srt.utils import get_compiler_backend
-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, suite="stage-b-kernel-benchmark-1-gpu-large")
+register_amd_ci(est_time=10, suite="jit-kernel-unit-test-amd")
 
 SIZE_LIST = get_benchmark_range(
     full_range=[2**n for n in range(4, 16)],  # 16 … 32K elements