From 46f85fc4bd309678e440201cfe8104b2c31c80d0 Mon Sep 17 00:00:00 2001
From: Erin Ho <14718778+hchings@users.noreply.github.com>
Date: Mon, 28 Apr 2025 20:10:55 +0000
Subject: [PATCH] waive test_attention_no_cache

Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com>
---
 tests/unittest/_torch/test_attention_no_cache.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/unittest/_torch/test_attention_no_cache.py b/tests/unittest/_torch/test_attention_no_cache.py
index 89de0147ae17..54e331a12025 100644
--- a/tests/unittest/_torch/test_attention_no_cache.py
+++ b/tests/unittest/_torch/test_attention_no_cache.py
@@ -182,6 +182,7 @@ def repeat_kv(hidden_states: torch.Tensor, n_rep: int) -> torch.Tensor:
 
 # skip for blackwell
 @skip_blackwell
+@pytest.mark.skip(reason="https://nvbugspro.nvidia.com/bug/5247232")
 # Convert parameterized tests to pytest parametrize
 @pytest.mark.parametrize("accuracy", [(1e-2, 1e-3)],
                          ids=lambda x: f"atol={x[0]} rtol={x[1]}")