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

more info in error msg.

上级 ff2f99f0
...@@ -253,7 +253,8 @@ class GpuDownsampleFactorMax(DownsampleFactorMax): ...@@ -253,7 +253,8 @@ class GpuDownsampleFactorMax(DownsampleFactorMax):
cudaError_t err = cudaGetLastError(); cudaError_t err = cudaGetLastError();
if( cudaSuccess != err) if( cudaSuccess != err)
{ {
PyErr_Format(PyExc_RuntimeError, "Cuda error: %%s: %%s.\\n", "kMaxPool_%(nodename)s", cudaGetErrorString(err)); PyErr_Format(PyExc_RuntimeError, "Cuda error: %%s: %%s.threads.x=%%d threads.y=%%d threads.z=%%d grid.x=%%d grid.y=%%d shared=%%d\\n", "kMaxPool_%(nodename)s",
cudaGetErrorString(err), block.x, block.y, block.z, grid.x, grid.y, shared);
%(fail)s; %(fail)s;
} }
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论