Allow a_cuda_ndarray.strides = [...]

theano/sandbox/cuda/cuda_ndarray.cu

@@ -2391,22 +2391,37 @@ CudaNdarray_get_strides(CudaNdarray *self, void *closure)
 static int
 CudaNdarray_set_strides(CudaNdarray *self, PyObject *value, void *closure)
 {
-    if (!PyTuple_Check(value)){
-        PyErr_SetString(PyExc_ValueError,
-                        "The new strides need to be encoded in a tupe");
-        return -1;
-    }
-    if (PyTuple_Size(value) != CudaNdarray_NDIM(self)){
+    //npy_intp newstrides_bytes[PyTuple_Size(value)];
+    if (PyTuple_Check(value)){
+        if (PyTuple_Size(value) != CudaNdarray_NDIM(self)){
+            PyErr_SetString(PyExc_ValueError,
+                            "The new strides tuple must have the same lenght"
+                            " as the number of dimensions");
+            return -1;
+        }
+    }else if (PyList_Check(value)){
+        if (PyList_Size(value) != CudaNdarray_NDIM(self)){
+            PyErr_SetString(PyExc_ValueError,
+                            "The new strides tuple must have the same lenght"
+                            " as the number of dimensions");
+            return -1;
+        }
+    }else{
         PyErr_SetString(PyExc_ValueError,
-                        "The new strides tuple must have the same lenght"
-                        " as the number of dimensions");
+                        "The new strides need to be encoded in a tuple or list");
         return -1;
     }
-    npy_intp newstrides[PyTuple_Size(value)];
-    //npy_intp newstrides_bytes[PyTuple_Size(value)];
-    for(int i=0; i < CudaNdarray_NDIM(self); i++){
-        newstrides[i] = PyInt_AsLong(PyTuple_GetItem(value, Py_ssize_t(i)));
-        //newstrides_bytes[i] = newstrides[i] * 4;
+    npy_intp newstrides[CudaNdarray_NDIM(self)];
+    if (PyTuple_Check(value)){
+        for(int i=0; i < CudaNdarray_NDIM(self); i++){
+            newstrides[i] = PyInt_AsLong(PyTuple_GetItem(value, Py_ssize_t(i)));
+            //newstrides_bytes[i] = newstrides[i] * 4;
+        }
+    }else if (PyList_Check(value)){
+        for(int i=0; i < CudaNdarray_NDIM(self); i++){
+            newstrides[i] = PyInt_AsLong(PyList_GetItem(value, Py_ssize_t(i)));
+            //newstrides_bytes[i] = newstrides[i] * 4;
+        }
     }
     /*
     // Don't do the check as ExtractDiag need that and NumPy seam to don't do

theano/sandbox/cuda/tests/test_cuda_ndarray.py

@@ -944,12 +944,16 @@ def test_base():
 
 def test_set_strides():
     a = cuda_ndarray.CudaNdarray.zeros((5, 5))
-    a.strides = (a.strides[1], a.strides[0])
-    try:
-        a.strides = [a.strides[1], a.strides[0]]
-        assert False
-    except ValueError:
-        pass
+
+    # Test with tuple
+    new_strides = (a.strides[1], a.strides[0])
+    a.strides = new_strides
+    assert a.strides == new_strides
+
+    # Test with list
+    new_strides = (a.strides[1], a.strides[0])
+    a.strides = [a.strides[1], a.strides[0]]
+    assert a.strides == new_strides
 
     try:
         a.strides = (a.strides[1],)