提交 33a2a9f2 authored 作者: James Bergstra's avatar James Bergstra

cuda - added special early-return for copying empty arrays to prevent division by zero.

上级 4ae437db
......@@ -1788,6 +1788,10 @@ int CudaNdarray_CopyFromCudaNdarray(CudaNdarray * self, CudaNdarray * other)
}
size *= (unsigned int) CudaNdarray_HOST_DIMS(self)[i];
}
if (0 == size)
{
return 0; //nothing to copy, we're done.
}
if (CudaNdarray_is_c_contiguous(self) && CudaNdarray_is_c_contiguous(other))
{
cublasScopy(size, CudaNdarray_DEV_DATA(other), 1, CudaNdarray_DEV_DATA(self), 1);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论