We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bfacf86 commit de83c49Copy full SHA for de83c49
tests/gpu/examples/test_lstm.py
@@ -150,6 +150,7 @@ def test_lstm_rnnt(self, dtype=torch.float):
150
self.assertEqual(h0.grad, h0_xpu.grad.cpu())
151
self.assertEqual(c0.grad, c0_xpu.grad.cpu())
152
153
+ @pytest.mark.skip("Unstable case. Will be fixed soon.")
154
def test_lstm_rnnt_onednn(self, dtype=torch.float):
155
with torch.xpu.compute_eng(torch.xpu.XPUComputeEng.ONEDNN):
156
rnn = nn.LSTM(240, 1024, num_layers=2)
0 commit comments