[UT] add skipif for rocm aiter sampler UT (#50530)
Signed-off-by: mayuyuace <qiming1.zhang@intel.com>
Qiming Zhang committed
Jul 31, 2026 at 01:32 UTC
5d7647a1092aba8eb4f41be91f89ef352854b318
1 file changed
+4
tests/v1/sample/test_topk_topp_sampler.py
+4
@@ -65,6 +65,10 @@ def test_sampler_threads_fp64_gumbel_to_topk_topp_sampler():
65
assert sampler.topk_topp_sampler.use_fp64_gumbel
66
67
68
+@pytest.mark.skipif(
69
+ not current_platform.is_rocm(),
70
+ reason="ROCm aiter sampler test only runs on ROCm",
71
+)
72
def test_rocm_aiter_sampler_defers_import_when_generators_force_native(
73
monkeypatch: pytest.MonkeyPatch,
74
):