Skip to content
This repository was archived by the owner on Nov 15, 2022. It is now read-only.

Commit 2f5dcfc

Browse files
Natalia Gimelsheinfacebook-github-bot
authored andcommitted
disable cpu half convolution test in nested tensor
Summary: T102092787 Reviewed By: mruberry Differential Revision: D31325281 fbshipit-source-id: 0fb8286a29309f8da9b02bca8322eeb7a9135ceb
1 parent a6faebf commit 2f5dcfc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_nested_tensor_functional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def test_conv2d_1x1_cuda(self):
8888
def test_conv2d_1x1_cpu(self):
8989
shapes = [(2, 2, 3), (2, 4, 2), (2, 2, 2)]
9090
weight = torch.randn(3*2*1*1).reshape(3, 2, 1, 1)
91-
self._test_conv2d_dtype(torch.float16, weight, torch.device('cpu'), shapes)
91+
# self._test_conv2d_dtype(torch.float16, weight, torch.device('cpu'), shapes)
9292
self._test_conv2d_dtype(torch.float32, weight, torch.device('cpu'), shapes)
9393

9494
@torch.inference_mode()

0 commit comments

Comments
 (0)