Fix compilation and test

theano/sandbox/gpuarray/dnn.py

@@ -127,9 +127,6 @@ def c_set_tensor4d(var, desc, err, fail):
   case GA_DOUBLE:
     dt = CUDNN_DATA_DOUBLE;
     break;
-  case GA_HALF:
-    dt = CUDNN_DATA_HALF;
-    break;
   default:
     PyErr_SetString(PyExc_TypeError, "Non-float datatype in c_set_tensor4d");
     return -1;

theano/sandbox/gpuarray/dnn_base.c

@@ -12,20 +12,23 @@ c_set_tensor4d(PyGpuArrayObject *var, cudnnTensorDescriptor_t desc) {
   case GA_DOUBLE:
     dt = CUDNN_DATA_DOUBLE;
     break;
-  case GA_HALF:
-    dt = CUDNN_DATA_HALF;
-    break;
   default:
     PyErr_SetString(PyExc_TypeError, "Non-float datatype in c_set_tensor4d");
     return -1;
   }
   ds = gpuarray_get_elsize(var->ga.typecode);
+
+  int str0, str1, str2, str3;
+  // cudnn do not like 0s in strides
+  str3 = PyGpuArray_STRIDES(var)[3]?PyGpuArray_STRIDES(var)[3]/ds:1;
+  str2 = PyGpuArray_STRIDES(var)[2]?PyGpuArray_STRIDES(var)[2]/ds:PyGpuArray_DIMS(var)[3];
+  str1 = PyGpuArray_STRIDES(var)[1]?PyGpuArray_STRIDES(var)[1]/ds:PyGpuArray_DIMS(var)[2]*PyGpuArray_DIMS(var)[3];
+  str0 = PyGpuArray_STRIDES(var)[0]?PyGpuArray_STRIDES(var)[0]/ds:PyGpuArray_DIMS(var)[2]*PyGpuArray_DIMS(var)[3]*PyGpuArray_DIMS(var)[1];
   cudnnStatus_t err = cudnnSetTensor4dDescriptorEx(
     desc, dt,
     PyGpuArray_DIM(var, 0), PyGpuArray_DIM(var, 1),
     PyGpuArray_DIM(var, 2), PyGpuArray_DIM(var, 3),
-    PyGpuArray_STRIDE(var, 0) / ds, PyGpuArray_STRIDE(var, 1) / ds,
-    PyGpuArray_STRIDE(var, 2) / ds, PyGpuArray_STRIDE(var, 3) / ds);
+    str0, str1, str2, str3);
   if (err != CUDNN_STATUS_SUCCESS) {
     PyErr_Format(PyExc_RuntimeError,
 		 "Could not set tensor4d descriptor: %s",
@@ -50,9 +53,6 @@ c_set_filter(PyGpuArrayObject *var, cudnnFilterDescriptor_t desc) {
   case GA_DOUBLE:
     dt = CUDNN_DATA_DOUBLE;
     break;
-  case GA_HALF:
-    dt = CUDNN_DATA_HALF;
-    break;
   default:
     PyErr_SetString(PyExc_TypeError, "Non-float datatype in c_set_filter");
     return -1;

theano/sandbox/gpuarray/tests/test_dnn.py

@@ -732,13 +732,11 @@ class test_SoftMax(test_nnet.test_SoftMax):
             x_gpu,
             f_z,
             f_gpu,
-            cmp,
-            self.mode,
+            cmp
         )
 
         self._test_softmax(
-            x, x, f_z, f_z, self._cmp,
-            self.mode
+            x, x, f_z, f_z, self._cmp
         )
 
         # Verify that the SoftmaxGrad -> Gpu[Dnn]SoftmaxGrad