提交 07eb476b authored 作者: nouiz's avatar nouiz

Merge pull request #3 from dwf/check_conv_const_shape_gpu

Spelling + spacing fixes
...@@ -84,7 +84,7 @@ CudaNdarray_conv_valid(const CudaNdarray *img, const CudaNdarray * kern, ...@@ -84,7 +84,7 @@ CudaNdarray_conv_valid(const CudaNdarray *img, const CudaNdarray * kern,
const int out_size_byte = out_size*sizeof(float); const int out_size_byte = out_size*sizeof(float);
if (!((THEANO_KERN_WID == CudaNdarray_HOST_DIMS(kern)[3]) || (THEANO_KERN_WID==0))){ if (!((THEANO_KERN_WID == CudaNdarray_HOST_DIMS(kern)[3]) || (THEANO_KERN_WID==0))){
PyErr_Format(PyExc_ValueError, "ERROR: This GpuConv code was compiled for" PyErr_Format(PyExc_ValueError, "ERROR: This GpuConv code was compiled for"
" %d kernel columns, but the kernel we received had %d colums!", " %d kernel columns, but the kernel we received had %d columns!",
THEANO_KERN_WID, CudaNdarray_HOST_DIMS(kern)[3]); THEANO_KERN_WID, CudaNdarray_HOST_DIMS(kern)[3]);
return -1; return -1;
} }
...@@ -768,7 +768,7 @@ CudaNdarray_conv_full(const CudaNdarray *img, const CudaNdarray * kern, CudaNdar ...@@ -768,7 +768,7 @@ CudaNdarray_conv_full(const CudaNdarray *img, const CudaNdarray * kern, CudaNdar
if (!((THEANO_KERN_WID == CudaNdarray_HOST_DIMS(kern)[3]) || (THEANO_KERN_WID==0))){ if (!((THEANO_KERN_WID == CudaNdarray_HOST_DIMS(kern)[3]) || (THEANO_KERN_WID==0))){
PyErr_Format(PyExc_ValueError, "ERROR: This GpuConv code was compiled for" PyErr_Format(PyExc_ValueError, "ERROR: This GpuConv code was compiled for"
" %d kernel columns, but the kernel we received had %d colums!", " %d kernel columns, but the kernel we received had %d columns!",
THEANO_KERN_WID, CudaNdarray_HOST_DIMS(kern)[3]); THEANO_KERN_WID, CudaNdarray_HOST_DIMS(kern)[3]);
return -1; return -1;
} }
......
...@@ -609,9 +609,10 @@ class TestConv2DGPU(unittest.TestCase): ...@@ -609,9 +609,10 @@ class TestConv2DGPU(unittest.TestCase):
if theano.config.mode in ['DebugMode', 'DEBUG_MODE']: if theano.config.mode in ['DebugMode', 'DEBUG_MODE']:
theano_mode = theano.compile.mode.get_mode('FAST_RUN').including('gpu') theano_mode = theano.compile.mode.get_mode('FAST_RUN').including('gpu')
for mode in ['valid', 'full']: for mode in ['valid', 'full']:
for shapes in [((3,2,8,8), (4,2,5,5), (8,8)), for shapes in [((3, 2, 8, 8), (4, 2, 5, 5), (8, 8)),
((3,2,8,8), (4,2,5,5), (5,8)), ((3, 2, 8, 8), (4, 2, 5, 5), (5, 8)),
#((3,2,8,8), (4,2,5,5), (8,5)),# We use only the number of columns. #((3, 2, 8, 8), (4, 2, 5, 5), (8, 5)),
# We use only the number of columns.
]: ]:
self.assertRaises(ValueError, _params_allgood, shapes[0], shapes[1], self.assertRaises(ValueError, _params_allgood, shapes[0], shapes[1],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论