提交 5ba85f6b authored 作者: Frederic's avatar Frederic

Better error message

上级 4ff5145e
...@@ -3679,7 +3679,11 @@ CudaNdarray_CopyFromArray(CudaNdarray * self, PyArrayObject*obj) ...@@ -3679,7 +3679,11 @@ CudaNdarray_CopyFromArray(CudaNdarray * self, PyArrayObject*obj)
CNDA_END_ALLOW_THREADS CNDA_END_ALLOW_THREADS
if (CUBLAS_STATUS_SUCCESS != cerr) if (CUBLAS_STATUS_SUCCESS != cerr)
{ {
PyErr_SetString(PyExc_RuntimeError, "error copying data to device memory"); PyErr_Format(PyExc_RuntimeError,
"CUBLAS error '%s' while copying %lli data element"
" to device memory",
cublasGetErrorString(cerr),
(long long)py_src_size);
Py_DECREF(py_src); Py_DECREF(py_src);
return -1; return -1;
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论