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

Convert an C assert to a Python error. A user got that.

上级 34e45a0c
...@@ -4069,7 +4069,16 @@ int CudaNdarray_CopyFromCudaNdarray(CudaNdarray * self, ...@@ -4069,7 +4069,16 @@ int CudaNdarray_CopyFromCudaNdarray(CudaNdarray * self,
}; break; }; break;
default: default:
{ {
assert (cudaSuccess == cudaGetLastError()); cudaError_t err = cudaGetLastError();
if(cudaSuccess != err){
PyErr_Format(PyExc_RuntimeError,
"Unexpected Cuda error: %s: %s\n",
"CudaNdarray_CopyFromCudaNdarray",
cudaGetErrorString(err));
Py_XDECREF(new_other);
return -1;
}
if (verbose) if (verbose)
fprintf(stderr, fprintf(stderr,
"Copying with default version unbroadcast=%d\n", "Copying with default version unbroadcast=%d\n",
...@@ -4092,7 +4101,7 @@ int CudaNdarray_CopyFromCudaNdarray(CudaNdarray * self, ...@@ -4092,7 +4101,7 @@ int CudaNdarray_CopyFromCudaNdarray(CudaNdarray * self,
CudaNdarray_DEV_DATA(self), CudaNdarray_DEV_DATA(self),
(const int *)CudaNdarray_DEV_STRIDES(self)); (const int *)CudaNdarray_DEV_STRIDES(self));
CNDA_THREAD_SYNC; CNDA_THREAD_SYNC;
cudaError_t err = cudaGetLastError(); err = cudaGetLastError();
if(verbose>1) if(verbose>1)
fprintf(stderr, fprintf(stderr,
"INFO k_elemwise_unary_rowmaj (n_blocks=%i," "INFO k_elemwise_unary_rowmaj (n_blocks=%i,"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论