Merge pull request #3794 from kmaehashi/bp-3790-v7-cudnn-v8

[backport] Temporary fixes for cudnn v8
cupy · Aug 17, 2020 · 53f2373 · 53f2373
2 parents db1b8db + 7ea5a0d
commit 53f2373
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/cupy_tests/test_cudnn.py b/tests/cupy_tests/test_cudnn.py
@@ -254,6 +254,12 @@ def setUp(self):
                 (ndim > 2 and deterministic and version < 6000) or
                 (ndim > 2 and deterministic and self.dtype == numpy.float64)):
             self.err = libcudnn.CuDNNError
+        elif (8000 <= version and
+              self.max_workspace_size == 0 and
+              int(cupy.cuda.device.get_compute_capability()) < 70 and
+              self.groups > 1 and ndim > 2 and
+              self.dtype == numpy.float16):
+            self.err = RuntimeError
         self._workspace_size = cudnn.get_max_workspace_size()
         cudnn.set_max_workspace_size(self.max_workspace_size)
 
@@ -327,6 +333,11 @@ def setUp(self):
                 (ndim > 2 and version < 6000) or
                 (ndim > 2 and self.dtype == numpy.float64)):
             self.err = libcudnn.CuDNNError
+        elif (8000 <= version and
+              int(cupy.cuda.device.get_compute_capability()) < 70 and
+              self.dilate > 1 and self.groups > 1 and ndim > 2 and
+              self.dtype == numpy.float16):
+            self.err = RuntimeError
         self._workspace_size = cudnn.get_max_workspace_size()
         cudnn.set_max_workspace_size(self.max_workspace_size)