提交 2dff4ecf authored 作者: nouiz's avatar nouiz

Merge pull request #509 from jaberg/gpu_conv_segfault_070312

failing GpuConv unittest
...@@ -705,3 +705,19 @@ def benchmark(): ...@@ -705,3 +705,19 @@ def benchmark():
exec_conv(version, shapes_valid, verbose, random, 'valid', print_=None, rtol=1e-3) exec_conv(version, shapes_valid, verbose, random, 'valid', print_=None, rtol=1e-3)
exec_conv(version, shapes_full, verbose, random, 'full') exec_conv(version, shapes_full, verbose, random, 'full')
def test_stack_rows_segfault_070312():
# 07/03/2012
# Running this unittest with cuda-memcheck exposes an illegal read.
# THEANO_FLAGS=device=gpu cuda-memcheck nosetests \
# test_conv_cuda_ndarray.py:test_stack_rows_segfault_070312
img = theano.shared(numpy.random.rand(1, 80, 96, 96).astype('float32'))
kern = theano.shared(numpy.random.rand(1, 80, 9, 9).astype('float32'))
out = theano.shared(numpy.random.rand(1, 2, 2, 3).astype('float32'))
op = theano.tensor.nnet.conv.ConvOp(imshp=(80, 96, 96), kshp=(9, 9),
nkern=1, bsize=1)
f = theano.function([], [], updates={out: op(img, kern)})
f()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论