Check if theta and grid are C-contiguous, and check grid dimensions

theano/gpuarray/c_code/spatialtf_grid.c

@@ -58,11 +58,7 @@ spatialtf_grid(PyGpuArrayObject * theta,
     }
 
     if ( NULL == *grid ||
-         PyGpuArray_NDIM( *grid ) != 4 ||
-         PyGpuArray_DIM( *grid, 0 ) != num_images ||
-         PyGpuArray_DIM( *grid, 1 ) != height ||
-         PyGpuArray_DIM( *grid, 2 ) != width  ||
-         PyGpuArray_DIM( *grid, 3) != 2 )
+         ! theano_size_check( *grid, 4, &(grid_dims[0]), (*grid)->ga.typecode ) )
     {
         Py_XDECREF( *grid );
 
@@ -80,6 +76,20 @@ spatialtf_grid(PyGpuArrayObject * theta,
         GpuArray_memset( &( (*grid)->ga ), 0 );
     }
 
+    if ( ! GpuArray_IS_C_CONTIGUOUS( &(theta->ga) ) )
+    {
+        PyErr_SetString( PyExc_MemoryError,
+                         "theta data is not C-contiguous" );
+        return -1;
+    }
+
+    if ( ! GpuArray_IS_C_CONTIGUOUS( &((*grid)->ga) ) )
+    {
+        PyErr_SetString( PyExc_MemoryError,
+                         "grid data is not C-contiguous" );
+        return -1;
+    }
+
     const void * theta_data = PyGpuArray_DEV_DATA( theta );
     void * grid_data = PyGpuArray_DEV_DATA( *grid );