From dd3d98a40d13fb1548517944893a8dec43d23982 Mon Sep 17 00:00:00 2001 From: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> Date: Fri, 24 Jul 2026 10:26:48 +0000 Subject: [PATCH] [https://nvbugs/6468821][infra] Split slow DGX B300 attention unit tests Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> --- tests/integration/test_lists/test-db/l0_dgx_b300.yml | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/integration/test_lists/test-db/l0_dgx_b300.yml b/tests/integration/test_lists/test-db/l0_dgx_b300.yml index 64f4d5632310..736319d8409f 100644 --- a/tests/integration/test_lists/test-db/l0_dgx_b300.yml +++ b/tests/integration/test_lists/test-db/l0_dgx_b300.yml @@ -15,7 +15,10 @@ l0_dgx_b300: stage: post_merge backend: pytorch tests: - - unittest/_torch/attention + - unittest/_torch/attention --ignore=unittest/_torch/attention/sparse/test_cute_dsl_fp8_paged_mqa_logits.py --ignore=unittest/_torch/attention/sparse/test_cute_dsl_fp4_paged_mqa_logits.py --ignore=unittest/_torch/attention/sparse/test_cute_dsl_gvr_topk_decode.py + - unittest/_torch/attention/sparse/test_cute_dsl_fp8_paged_mqa_logits.py + - unittest/_torch/attention/sparse/test_cute_dsl_fp4_paged_mqa_logits.py + - unittest/_torch/attention/sparse/test_cute_dsl_gvr_topk_decode.py - unittest/_torch/executor # ------------- modules (multi-GPU) --------------- - unittest/_torch/modules/test_mla_helix.py