提交 053c63fb authored 作者: Dustin Webb's avatar Dustin Webb

Fixed API compatibility issues and simplified memset call.

上级 354169fc
......@@ -403,15 +403,15 @@ def gemv_c_code(aa, xx, yy, zz, alpha, beta, destructive, fail, force_init_beta=
}
else if (%(force_init_beta)d)
{
if (PyArray_IS_C_CONTIGUOUS(%(zz)s))
if (PyArray_CHKFLAGS(%(zz)s, NPY_ARRAY_C_CONTIGUOUS))
{
if (PyArray_DESCR(%(zz)s)->type_num == NPY_FLOAT)
{
memset((void *)PyArray_DATA(%(zz)s), 0, PyArray_SIZE(%(zz)s)*sizeof(float));
memset((void *)PyArray_DATA(%(zz)s), 0, PyArray_ITEMSIZE(%(zz)s));
}
else if (PyArray_DESCR(%(zz)s)->type_num == NPY_DOUBLE)
{
memset((void *)PyArray_DATA(%(zz)s), 0, PyArray_SIZE(%(zz)s)*sizeof(double));
memset((void *)PyArray_DATA(%(zz)s), 0, PyArray_ITEMSIZE(%(zz)s));
}
else
{
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论