diff --git a/test/registered/jit/diffusion/test_ltx2_ada_values.py b/test/registered/jit/diffusion/test_ltx2_ada_values.py index 8ae4c4a30..73721e829 100644 --- a/test/registered/jit/diffusion/test_ltx2_ada_values.py +++ b/test/registered/jit/diffusion/test_ltx2_ada_values.py @@ -4,9 +4,10 @@ import pytest import torch from sglang.jit_kernel.diffusion.triton.ltx2_ada_values import ltx2_ada_values9 -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=8, stage="base-b-kernel-unit", runner_config="1-gpu-large") +register_amd_ci(est_time=8, suite="nightly-amd-kernel-1-gpu", nightly=True) DEVICE = "cuda"