提交 94a43515 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix spelling of CUBLAS_STATUS_INVALID_VALUE.

上级 84161c84
......@@ -446,7 +446,7 @@ static cublasStatus_t SgerBatched(cublasHandle_t handle, int m, int n,
}
if (grid.x * grid.y * grid.z > 65535) {
if (grid.x * grid.y > 65535)
return CUBLAS_INVALID_VALUE;
return CUBLAS_STATUS_INVALID_VALUE;
grid.z = (65535 / (grid.x * grid.y));
}
cublasGetPointerMode(handle, &mode);
......@@ -592,7 +592,7 @@ CudaNdarray_HOST_STRIDES(%(out)s)[0], CudaNdarray_HOST_STRIDES(%(out)s)[1],
CudaNdarray_HOST_DIMS(%(x)s)[1] *
CudaNdarray_HOST_DIMS(%(y)s)[1]);
if (err != CUBLAS_STATUS_SUCCESS) {
if (err == CUBLAS_INVALID_VALUE) {
if (err == CUBLAS_STATUS_INVALID_VALUE) {
/* The current code would be much too slow for sizes any larger
than this. */
PyErr_SetString(PyExc_ValueError,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论