summaryrefslogtreecommitdiff
path: root/test/test_nn.py
diff options
context:
space:
mode:
authorJunjie Bai <bai@in.tum.de>2019-02-13 17:12:01 -0800
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>2019-02-13 17:18:47 -0800
commit9b7f3da74bd782f1cc9933da5e4a03825af8e046 (patch)
treea86a31852fa649622a6acf42108e59736f2a533e /test/test_nn.py
parenta670824fee57a2b261168a2bfcfc1f4e008d8d4a (diff)
downloadpytorch-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.py1
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.