From bf16a6bc3c515c6713f25a4aa08c4d3f67eed780 Mon Sep 17 00:00:00 2001 From: Junjie Bai Date: Sat, 16 Feb 2019 18:00:10 -0800 Subject: [PATCH] Skip onnx logsoftmax tests in rocm (#17170) Summary: similar to softmax there are issues of getting nan randomly Pull Request resolved: https://github.com/pytorch/pytorch/pull/17170 Differential Revision: D14110515 Pulled By: bddppq fbshipit-source-id: 5c97661184d45a02122fd69d35a839fdf4520c8c --- caffe2/python/onnx/tests/onnx_backend_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/caffe2/python/onnx/tests/onnx_backend_test.py b/caffe2/python/onnx/tests/onnx_backend_test.py index 11392ba..2a600be 100644 --- a/caffe2/python/onnx/tests/onnx_backend_test.py +++ b/caffe2/python/onnx/tests/onnx_backend_test.py @@ -79,6 +79,7 @@ if 'JENKINS_URL' in os.environ: if workspace.has_hip_support: # TODO: Investigate flakiness in ROCM Softmax (it sometimes give NaN). backend_test.exclude(r'test_softmax_.*_cuda') + backend_test.exclude(r'test_logsoftmax_.*_cuda') # import all test cases at global scope to make them visible to python.unittest globals().update(backend_test -- 2.7.4