提交 af8ef4b5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix compilation crash show by…

Fix compilation crash show by THEANO_FLAGS='floatX=float32,dnn.conv.algo_bwd_data=time_once,dnn.conv.algo_bwd_filter=time_once' theano-nose theano/sandbox/cuda/tests/test_dnn.py:test_dnn_conv_grad. fix gh-5517
上级 8757c2dc
...@@ -165,8 +165,10 @@ APPLY_SPECIFIC(conv_gi)(PyGpuArrayObject *kerns, PyGpuArrayObject *output, ...@@ -165,8 +165,10 @@ APPLY_SPECIFIC(conv_gi)(PyGpuArrayObject *kerns, PyGpuArrayObject *output,
} }
err = cudnnFindConvolutionBackwardDataAlgorithmEx( err = cudnnFindConvolutionBackwardDataAlgorithmEx(
_handle, APPLY_SPECIFIC(kerns), APPLY_SPECIFIC(output), desc, _handle, APPLY_SPECIFIC(kerns), PyGpuArray_DEV_DATA(kerns),
APPLY_SPECIFIC(input), 1, &count, &choice, *(void **)tmpmem, free); APPLY_SPECIFIC(output), PyGpuArray_DEV_DATA(output), desc,
APPLY_SPECIFIC(input), PyGpuArray_DEV_DATA(input),
1, &count, &choice, *(void **)tmpmem, free);
gpudata_release(tmpmem); gpudata_release(tmpmem);
if (err != CUDNN_STATUS_SUCCESS) { if (err != CUDNN_STATUS_SUCCESS) {
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论