From 2582134a59480e6b8aaff48f6bef09d4d54fa17b Mon Sep 17 00:00:00 2001 From: "Hsiu-Chun, Hung" <160560375+Emmanuel0612@users.noreply.github.com> Date: Tue, 2 Jun 2026 15:24:59 +0800 Subject: [PATCH] [AMD] Add amd ci mamba state scatter test (#26677) Co-authored-by: Hung --- test/registered/unit/layers/test_mamba_state_scatter_triton.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/registered/unit/layers/test_mamba_state_scatter_triton.py b/test/registered/unit/layers/test_mamba_state_scatter_triton.py index b02136462..908149f62 100644 --- a/test/registered/unit/layers/test_mamba_state_scatter_triton.py +++ b/test/registered/unit/layers/test_mamba_state_scatter_triton.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=7, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=7, suite="stage-b-test-1-gpu-small-amd-mi35x") import os import unittest