Merge pull request #6026 from notoraptor/re-allow-runtime-algos-cudnn-conv3d

theano/gpuarray/dnn.py

@@ -559,7 +559,8 @@ class GpuDnnConv(DnnBase):
             raise TypeError("The number of dimensions of "
                             "img, kern and output must match")
 
-        if img.type.ndim == 5 and self.algo not in cudnn.conv3d_fwd_algorithms:
+        if img.type.ndim == 5 and self.algo not in (cudnn.conv3d_fwd_algorithms +
+                                                    SUPPORTED_DNN_CONV_ALGO_RUNTIME):
             raise ValueError("convolution algo %s can't be used for "
                              "3d convolutions", (self.algo,))
 
@@ -729,7 +730,8 @@ class GpuDnnConvGradW(DnnBase):
             raise TypeError("The number of dimensions of "
                             "img, topgrad and output must match")
 
-        if img.type.ndim == 5 and self.algo not in cudnn.conv3d_bwd_filter_algorithms:
+        if img.type.ndim == 5 and self.algo not in (cudnn.conv3d_bwd_filter_algorithms +
+                                                    SUPPORTED_DNN_CONV_ALGO_RUNTIME):
             raise ValueError("convolution algo %s can't be used for "
                              "3d convolutions", (self.algo,))
 
@@ -832,7 +834,8 @@ class GpuDnnConvGradI(DnnBase):
             raise TypeError("The number of dimensions of "
                             "kern, topgrad and output must match")
 
-        if kern.type.ndim == 5 and self.algo not in cudnn.conv3d_bwd_data_algorithms:
+        if kern.type.ndim == 5 and self.algo not in (cudnn.conv3d_bwd_data_algorithms +
+                                                     SUPPORTED_DNN_CONV_ALGO_RUNTIME):
             raise ValueError("convolution algo %s can't be used for "
                              "3d convolutions", (self.algo,))