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

remove compiler warning and better error msg.

上级 67e47e63
...@@ -22,8 +22,8 @@ void * device_malloc(size_t size) ...@@ -22,8 +22,8 @@ void * device_malloc(size_t size)
void * rval=NULL; void * rval=NULL;
if (cudaSuccess != cudaMalloc(&rval, size)) if (cudaSuccess != cudaMalloc(&rval, size))
{ {
fprintf(stderr, "Error allocating %i bytes of device memory.\n", size); fprintf(stderr, "Error allocating %li bytes of device memory.\n", (long)size);
PyErr_Format(PyExc_MemoryError, "error allocating %i bytes of device memory", size); PyErr_Format(PyExc_MemoryError, "error allocating %li bytes of device memory", (long)size);
return NULL; return NULL;
} }
_outstanding_mallocs[0] += (rval != NULL); _outstanding_mallocs[0] += (rval != NULL);
...@@ -1974,9 +1974,10 @@ int CudaNdarray_gemm(float alpha, const CudaNdarray * A, const CudaNdarray * B, ...@@ -1974,9 +1974,10 @@ int CudaNdarray_gemm(float alpha, const CudaNdarray * A, const CudaNdarray * B,
return -1; return -1;
}; };
CNDA_THREAD_SYNC; CNDA_THREAD_SYNC;
if (CUBLAS_STATUS_SUCCESS != cublasGetError()) cudaError_t err = cudaGetLastError();
if (CUBLAS_STATUS_SUCCESS != err)
{ {
PyErr_SetString(PyExc_RuntimeError, "cublassGemm failed"); PyErr_Format(PyExc_RuntimeError, "cublassGemm failed (%s)",cudaGetErrorString(err));
return -1; return -1;
} }
return 0; return 0;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论