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

code simplification

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