[AMD][CI] Skip failing Wave test and relax multi-LoRA output check (#38585)
This commit is contained in:
@@ -283,6 +283,9 @@ class TestWaveAttention(unittest.TestCase):
|
||||
self.assertTrue(cos_sim.item() > 0.99)
|
||||
self.assertTrue(torch.allclose(o, o_triton, atol=3e-2))
|
||||
|
||||
@unittest.skip(
|
||||
"Wave grouped decode is temporarily disabled because it produces NaNs on ROCm 10."
|
||||
)
|
||||
def test_grouped_decode_attention(self):
|
||||
seq_lens = [5, 100, 128, 500]
|
||||
configs = [
|
||||
|
||||
Reference in New Issue
Block a user