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

Revert "[ROCm] Enabling several UTs (#161715)"

Browse Source 
This reverts commit b9ba612f7a968f7b27e121ca8f4d0a4d954f5354.

Reverted https://github.com/pytorch/pytorch/pull/161715 on behalf of https://github.com/jeanschmidt due to Need to revert in order to revert https://github.com/pytorch/pytorch/pull/159473, feel free to merge it back once conflicts are cleared ([comment](https://github.com/pytorch/pytorch/pull/161715#issuecomment-3264040604))
This commit is contained in:
PyTorch MergeBot
2025-09-07 21:03:17 +00:00
parent e246a85b76
commit 8235c4f65d
25 changed files with 82 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_segment_reductions.py
View File

@ -14,6 +14,7 @@ from torch.testing._internal.common_utils import (
run_tests,
gradcheck,
parametrize,
skipIfRocm,
)
@ -230,6 +231,7 @@ class TestSegmentReductions(TestCase):
length_type,
)
@skipIfRocm
@dtypes(
*product(
(torch.half, torch.bfloat16, torch.float, torch.double),