提交 a29d0982 authored 作者: Frederic's avatar Frederic

code simplification

上级 865ea186
......@@ -1100,7 +1100,7 @@ class Gemm(GemmRelated):
dims[0] = PyArray_DIMS(%(_z)s)[0];
dims[1] = PyArray_DIMS(%(_z)s)[1];
%(_zout)s = (PyArrayObject*)PyArray_SimpleNew(2, dims,
PyArray_TYPE((PyArrayObject*) py_%(_z)s));
PyArray_TYPE(%(_z)s));
//fprintf(stderr, "Gemm Allocating %%i %%i\\n", dims[0], dims[1]);
if(!%(_zout)s) {
PyErr_SetString(PyExc_MemoryError,
......@@ -1188,7 +1188,7 @@ class Gemm(GemmRelated):
def c_code_cache_version(self):
gv = self.build_gemm_version()
if gv:
return (4,) + gv
return (5,) + gv
else:
return gv
......
......@@ -91,7 +91,7 @@ def ger_c_code(A, a, x, y, Z, destructive, fail):
|| ((PyArray_STRIDES(%(Z)s)[0] != elemsize)
&& (PyArray_STRIDES(%(Z)s)[1] != elemsize)))
{
if (%(Z)s) Py_XDECREF(%(Z)s);
Py_XDECREF(%(Z)s);
%(Z)s = (PyArrayObject*) PyArray_SimpleNew(2, dims,
PyArray_TYPE(%(A)s));
if(!%(Z)s) {
......@@ -324,7 +324,7 @@ class CGer(BaseBLAS, Ger):
return code
def c_code_cache_version(self):
return (9, blas_header_version())
return (10, blas_header_version())
cger_inplace = CGer(True)
cger_no_inplace = CGer(False)
......@@ -425,9 +425,9 @@ def gemv_c_code(aa, xx, yy, zz, alpha, beta, destructive, fail,
if ((NULL == %(zz)s)
|| (PyArray_DIMS(%(zz)s)[0] != PyArray_DIMS(%(aa)s)[0]))
{
if (%(zz)s) Py_XDECREF(%(zz)s);
Py_XDECREF(%(zz)s);
%(zz)s = (PyArrayObject*)PyArray_SimpleNew(1,
PyArray_DIMS(%(aa)s), PyArray_TYPE((PyArrayObject*) py_%(aa)s));
PyArray_DIMS(%(aa)s), PyArray_TYPE(%(aa)s));
if(!%(zz)s) {
PyErr_SetString(PyExc_MemoryError,
"failed to alloc gemv output");
......@@ -695,7 +695,7 @@ class CGemv(BaseBLAS, Gemv):
return code
def c_code_cache_version(self):
return (11, blas_header_version())
return (12, blas_header_version())
cgemv_inplace = CGemv(inplace=True)
cgemv_no_inplace = CGemv(inplace=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论