From 357a3038599bc72cf53e57b662c3658343ea8abc Mon Sep 17 00:00:00 2001 From: longcheng-nv <243710427+longcheng-nv@users.noreply.github.com> Date: Wed, 16 Sep 2026 12:21:31 +0000 Subject: [PATCH] [None][test] Gate SM103 graph test on SM count Signed-off-by: longcheng-nv <243710427+longcheng-nv@users.noreply.github.com> --- .../_torch/thop/parallel/test_gvr_selfsampling_topk.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/unittest/_torch/thop/parallel/test_gvr_selfsampling_topk.py b/tests/unittest/_torch/thop/parallel/test_gvr_selfsampling_topk.py index e5b72c91ba49..dbf75afa7391 100644 --- a/tests/unittest/_torch/thop/parallel/test_gvr_selfsampling_topk.py +++ b/tests/unittest/_torch/thop/parallel/test_gvr_selfsampling_topk.py @@ -765,9 +765,11 @@ def test_selfsampling_route_sm103_large_batch_k2048_register_rung() -> None: @pytest.mark.skipif(getSMVersion() != 103, reason="requires an SM103 GPU") def test_selfsampling_sm103_b512_k2048_cuda_graph() -> None: - """Exercise the B300-only register plan through warmup and graph replay.""" + """Exercise the 148-SM SM103 register plan through warmup and graph replay.""" rows, n, npad, top_k = 512, 4111, 4160, 2048 properties = torch.cuda.get_device_properties(torch.cuda.current_device()) + if properties.multi_processor_count != 148: + pytest.skip("requires an SM103 GPU with 148 SMs") plan = ss_host.route( rows, n,