change dtype for better testing

pytorch · Oct 29, 2020 · fedbb47 · fedbb47
1 parent f0876e9
commit fedbb47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/onnx/test_pytorch_onnx_onnxruntime.py b/test/onnx/test_pytorch_onnx_onnxruntime.py
@@ -2362,7 +2362,7 @@ def test_logsoftmax_dim(self):
     def test_logsoftmax_dtype(self):
         class Model(torch.nn.Module):
             def forward(self, x):
-                return torch.nn.functional.log_softmax(x, dim=1, dtype=torch.float32)
+                return torch.nn.functional.log_softmax(x, dim=1, dtype=torch.float64)
 
         x = torch.randn(3, 4, 5, requires_grad=True)
         self.run_test(Model(), x)