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

crash fix. Add the missing inputs to the function call. It also need the real…

crash fix. Add the missing inputs to the function call. It also need the real data. I don't sync as we don't care of the values.
上级 94af282e
...@@ -121,10 +121,12 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns, ...@@ -121,10 +121,12 @@ APPLY_SPECIFIC(conv_fwd)(PyGpuArrayObject *input, PyGpuArrayObject *kerns,
PyErr_SetString(PyExc_MemoryError, "Could not allocate working GPU memory"); PyErr_SetString(PyExc_MemoryError, "Could not allocate working GPU memory");
return -1; return -1;
} }
// We don't sync the buffer as we don't care about the values.
err = cudnnFindConvolutionForwardAlgorithmEx( err = cudnnFindConvolutionForwardAlgorithmEx(
_handle, APPLY_SPECIFIC(input), APPLY_SPECIFIC(kerns), _handle, APPLY_SPECIFIC(input), PyGpuArray_DEV_DATA(input),
desc, APPLY_SPECIFIC(output), 1, &count, &choice, *(void **)tmpmem, APPLY_SPECIFIC(kerns), PyGpuArray_DEV_DATA(kerns),
desc, APPLY_SPECIFIC(output), PyGpuArray_DEV_DATA(*output),
1, &count, &choice, *(void **)tmpmem,
free); free);
gpudata_release(tmpmem); gpudata_release(tmpmem);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论