OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ROCm] skip test_RNN_dropout_state (#149446)

Browse Source 
PR to skip test_nn.py::TestNN::test_RNN_dropout_state
Currently ROCm doesn't support dropout value for RNN

PR to enable RNN dropout on ROCm still in review and blocked pytorch/pytorch#144572

Fixes: https://github.com/pytorch/pytorch/issues/68849

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149446
Approved by: https://github.com/pruthvistony, https://github.com/jeffdaily
This commit is contained in:
Dmitry Nikolaev 2025-03-20 17:22:39 +00:00 committed by PyTorch MergeBot
parent 1d9401befc
commit c99efc08fb
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_nn.py
View File

@ -4375,6 +4375,8 @@ tensor(..., device='meta', size=(1,), requires_grad=True)""")
outs[0].sum().backward()
@unittest.skipIf(not TEST_CUDNN, "needs cudnn")
@skipIfRocm(msg="ROCm doesn't support 'dropout' for RNN "
"(WIP to enable dropout https://github.com/pytorch/pytorch/pull/144572)")
def test_RNN_dropout_state(self):
for p in (0, 0.1234):
for train in (True, False):