frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] skip test_sparse_csr test_triton_bsr_softmax_cuda (#118006)

Browse Source 
The tests were taking too long and leading to CI timeouts.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118006
Approved by: https://github.com/huydhn
This commit is contained in:
Jeff Daily
2024-01-23 00:09:42 +00:00
committed by PyTorch MergeBot
parent c6be5d55a5
commit a27a6e8cf1
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_sparse_csr.py
View File

@ -3463,6 +3463,7 @@ class TestSparseCompressedTritonKernels(TestCase):
return d
@onlyCUDA
@skipIfRocm(msg="test is too slow on ROCm stack")
@dtypes(torch.half, torch.bfloat16, torch.float)
@dtypesIfCUDA(torch.half, *[torch.bfloat16] if SM80OrLater else [], torch.float)
@unittest.skipIf(IS_FBCODE and IS_REMOTE_GPU, "Test requires Triton")