diff options
author | Junjie Bai <bai@in.tum.de> | 2019-02-13 17:12:01 -0800 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-02-13 17:18:47 -0800 |
commit | 9b7f3da74bd782f1cc9933da5e4a03825af8e046 (patch) | |
tree | a86a31852fa649622a6acf42108e59736f2a533e /test/test_nn.py | |
parent | a670824fee57a2b261168a2bfcfc1f4e008d8d4a (diff) | |
download | pytorch-9b7f3da74bd782f1cc9933da5e4a03825af8e046.tar.gz pytorch-9b7f3da74bd782f1cc9933da5e4a03825af8e046.tar.bz2 pytorch-9b7f3da74bd782f1cc9933da5e4a03825af8e046.zip |
Skip test_cudnn_multiple_threads_same_device on ROCm (flaky) (#17061)
Summary:
cc iotamudelta
https://ci.pytorch.org/jenkins/job/pytorch-builds/job/py2-clang7-rocmdeb-ubuntu16.04-test/10722//console
https://ci.pytorch.org/jenkins/job/pytorch-builds/job/py2-clang7-rocmdeb-ubuntu16.04-test/10710//console
https://ci.pytorch.org/jenkins/job/pytorch-builds/job/py2-clang7-rocmdeb-ubuntu16.04-test/10753//console
https://ci.pytorch.org/jenkins/job/pytorch-builds/job/py2-devtoolset7-rocmrpm-centos7.5-test/1756//console
```
19:07:18 ======================================================================
19:07:18 FAIL: test_cudnn_multiple_threads_same_device (test_nn.TestNN)
19:07:18 ----------------------------------------------------------------------
19:07:18 Traceback (most recent call last):
19:07:18 File "/var/lib/jenkins/workspace/test/test_nn.py", line 3905, in test_cudnn_multiple_threads_same_device
19:07:18 (2048 - test_iters) * (2048 - test_iters))
19:07:18 File "/var/lib/jenkins/workspace/test/common_utils.py", line 453, in assertEqual
19:07:18 super(TestCase, self).assertLessEqual(abs(x - y), prec, message)
19:07:18 AssertionError: 3794704.0 not less than or equal to 1e-05 :
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17061
Differential Revision: D14069324
Pulled By: bddppq
fbshipit-source-id: e33b09abca217a62a8b577f9c332ea22985ef4ff
Diffstat (limited to 'test/test_nn.py')
-rw-r--r-- | test/test_nn.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/test/test_nn.py b/test/test_nn.py index 5adf5b5a24..c3761bcfb5 100644 --- a/test/test_nn.py +++ b/test/test_nn.py @@ -3858,6 +3858,7 @@ class TestNN(NNTestCase): @unittest.skipIf(not TEST_CUDA, 'CUDA not available') @unittest.skipIf(not TEST_CUDNN, 'CUDNN not available') + @skipIfRocm def test_cudnn_multiple_threads_same_device(self): # This function is intended to test the lazy creation and reuse of per-thread # cudnn handles on each device in aten/src/ATen/cudnn/Handles.cpp. |