提交 b7a9bc59 authored 作者: Frederic's avatar Frederic

More detailed error message.

上级 5f2ed7aa
...@@ -549,7 +549,7 @@ class GpuSoftmaxWithBias(GpuOp): ...@@ -549,7 +549,7 @@ class GpuSoftmaxWithBias(GpuOp):
def c_code_cache_version(self): def c_code_cache_version(self):
#return () #return ()
return (8,) + inline_softmax.code_version return (9,) + inline_softmax.code_version
def c_code(self, node, nodename, inp, out, sub): def c_code(self, node, nodename, inp, out, sub):
x, b = inp x, b = inp
...@@ -649,9 +649,11 @@ class GpuSoftmaxWithBias(GpuOp): ...@@ -649,9 +649,11 @@ class GpuSoftmaxWithBias(GpuOp):
if( cudaSuccess != err) if( cudaSuccess != err)
{ {
PyErr_Format(PyExc_RuntimeError, PyErr_Format(PyExc_RuntimeError,
"Cuda error: %%s: %%s.\\n", "Cuda error: %%s: %%s. n_blocks=%%d,"
" n_threads=%%d, n_shared_bytes=%%d\\n",
"kSoftmaxWithBias_%(nodename)s", "kSoftmaxWithBias_%(nodename)s",
cudaGetErrorString(err)); cudaGetErrorString(err),
n_blocks, n_threads, n_shared_bytes);
%(fail)s; %(fail)s;
} }
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论