Skip to content

Commit 6aa5dce

Browse files
authored
only skip fp32 for conv3d training (same with 1.12 release) (#1197)
1 parent 8ab8324 commit 6aa5dce

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/cpu/test_weight_prepack.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,9 +237,9 @@ def test_conv2d_training(self):
237237

238238
# TODO: add inference case.
239239

240-
@unittest.skip("Stock Pytorch does not support channelslast for Conv3d")
241240
def test_conv3d_training(self):
242-
self._test_convolution_training_base(dim=3, dtype=torch.float)
241+
# Stock Pytorch does not support channelslast for Conv3d
242+
# self._test_convolution_training_base(dim=3, dtype=torch.float)
243243
if core.onednn_has_bf16_support():
244244
self._test_convolution_training_base(dim=3, dtype=torch.bfloat16, rtol=1e-2, atol=1e-03)
245245
# TODO: add inference case.

0 commit comments

Comments
 (0)