Generated by Cython 0.29.28
Yellow lines hint at Python interaction.
Click on a line that starts with a "+
" to see the C code that Cython generated for it.
Raw output: cg_ext.c
+01: #cython: wraparound=False, boundscheck=False, cdivision=True, profile=False, nonecheck=False, overflowcheck=False, cdivision_warnings=False, unraisable_tracebacks=False
__pyx_t_1 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 1, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_test, __pyx_t_1) < 0) __PYX_ERR(0, 1, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
02: import cython
03:
04: # import both numpy and the Cython declarations for numpy
+05: import numpy as np
__pyx_t_1 = __Pyx_Import(__pyx_n_s_numpy, 0, -1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 5, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_np, __pyx_t_1) < 0) __PYX_ERR(0, 5, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
06: cimport numpy as np
07:
08: # declare the interface to the C code
09: cdef extern from "cg.c":
10: int _jacobi_precon_c(double* data, long* colind, long* row_ptr, double* precon, int M)
11: int _cg_solve_c(double* data, long* colind, long* row_ptr, double* b, double* x, int imax, double tol, double a_tol, int M)
12: int _cg_solve_c_precon(double* data, long* colind, long* row_ptr, double* b, double* x, int imax, double tol, double a_tol, int M, double* precon)
13:
+14: def jacobi_precon_c(object csr_sparse, np.ndarray[double, ndim=1, mode="c"] precon not None):
/* Python wrapper */ static PyObject *__pyx_pw_5anuga_9utilities_6cg_ext_1jacobi_precon_c(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/ static PyMethodDef __pyx_mdef_5anuga_9utilities_6cg_ext_1jacobi_precon_c = {"jacobi_precon_c", (PyCFunction)(void*)(PyCFunctionWithKeywords)__pyx_pw_5anuga_9utilities_6cg_ext_1jacobi_precon_c, METH_VARARGS|METH_KEYWORDS, 0}; static PyObject *__pyx_pw_5anuga_9utilities_6cg_ext_1jacobi_precon_c(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) { PyObject *__pyx_v_csr_sparse = 0; PyArrayObject *__pyx_v_precon = 0; PyObject *__pyx_r = 0; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("jacobi_precon_c (wrapper)", 0); { static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_csr_sparse,&__pyx_n_s_precon,0}; PyObject* values[2] = {0,0}; if (unlikely(__pyx_kwds)) { Py_ssize_t kw_args; const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args); switch (pos_args) { case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1); CYTHON_FALLTHROUGH; case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0); CYTHON_FALLTHROUGH; case 0: break; default: goto __pyx_L5_argtuple_error; } kw_args = PyDict_Size(__pyx_kwds); switch (pos_args) { case 0: if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_csr_sparse)) != 0)) kw_args--; else goto __pyx_L5_argtuple_error; CYTHON_FALLTHROUGH; case 1: if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_precon)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("jacobi_precon_c", 1, 2, 2, 1); __PYX_ERR(0, 14, __pyx_L3_error) } } if (unlikely(kw_args > 0)) { if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "jacobi_precon_c") < 0)) __PYX_ERR(0, 14, __pyx_L3_error) } } else if (PyTuple_GET_SIZE(__pyx_args) != 2) { goto __pyx_L5_argtuple_error; } else { values[0] = PyTuple_GET_ITEM(__pyx_args, 0); values[1] = PyTuple_GET_ITEM(__pyx_args, 1); } __pyx_v_csr_sparse = values[0]; __pyx_v_precon = ((PyArrayObject *)values[1]); } goto __pyx_L4_argument_unpacking_done; __pyx_L5_argtuple_error:; __Pyx_RaiseArgtupleInvalid("jacobi_precon_c", 1, 2, 2, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 14, __pyx_L3_error) __pyx_L3_error:; __Pyx_AddTraceback("anuga.utilities.cg_ext.jacobi_precon_c", __pyx_clineno, __pyx_lineno, __pyx_filename); __Pyx_RefNannyFinishContext(); return NULL; __pyx_L4_argument_unpacking_done:; if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_precon), __pyx_ptype_5numpy_ndarray, 0, "precon", 0))) __PYX_ERR(0, 14, __pyx_L1_error) __pyx_r = __pyx_pf_5anuga_9utilities_6cg_ext_jacobi_precon_c(__pyx_self, __pyx_v_csr_sparse, __pyx_v_precon); int __pyx_lineno = 0; const char *__pyx_filename = NULL; int __pyx_clineno = 0; /* function exit code */ goto __pyx_L0; __pyx_L1_error:; __pyx_r = NULL; __pyx_L0:; __Pyx_RefNannyFinishContext(); return __pyx_r; } static PyObject *__pyx_pf_5anuga_9utilities_6cg_ext_jacobi_precon_c(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csr_sparse, PyArrayObject *__pyx_v_precon) { int __pyx_v_M; CYTHON_UNUSED int __pyx_v_err; PyArrayObject *__pyx_v_data = 0; PyArrayObject *__pyx_v_colind = 0; PyArrayObject *__pyx_v_row_ptr = 0; __Pyx_LocalBuf_ND __pyx_pybuffernd_colind; __Pyx_Buffer __pyx_pybuffer_colind; __Pyx_LocalBuf_ND __pyx_pybuffernd_data; __Pyx_Buffer __pyx_pybuffer_data; __Pyx_LocalBuf_ND __pyx_pybuffernd_precon; __Pyx_Buffer __pyx_pybuffer_precon; __Pyx_LocalBuf_ND __pyx_pybuffernd_row_ptr; __Pyx_Buffer __pyx_pybuffer_row_ptr; PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("jacobi_precon_c", 0); __pyx_pybuffer_data.pybuffer.buf = NULL; __pyx_pybuffer_data.refcount = 0; __pyx_pybuffernd_data.data = NULL; __pyx_pybuffernd_data.rcbuffer = &__pyx_pybuffer_data; __pyx_pybuffer_colind.pybuffer.buf = NULL; __pyx_pybuffer_colind.refcount = 0; __pyx_pybuffernd_colind.data = NULL; __pyx_pybuffernd_colind.rcbuffer = &__pyx_pybuffer_colind; __pyx_pybuffer_row_ptr.pybuffer.buf = NULL; __pyx_pybuffer_row_ptr.refcount = 0; __pyx_pybuffernd_row_ptr.data = NULL; __pyx_pybuffernd_row_ptr.rcbuffer = &__pyx_pybuffer_row_ptr; __pyx_pybuffer_precon.pybuffer.buf = NULL; __pyx_pybuffer_precon.refcount = 0; __pyx_pybuffernd_precon.data = NULL; __pyx_pybuffernd_precon.rcbuffer = &__pyx_pybuffer_precon; { __Pyx_BufFmt_StackElem __pyx_stack[1]; if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_precon.rcbuffer->pybuffer, (PyObject*)__pyx_v_precon, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 14, __pyx_L1_error) } __pyx_pybuffernd_precon.diminfo[0].strides = __pyx_pybuffernd_precon.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_precon.diminfo[0].shape = __pyx_pybuffernd_precon.rcbuffer->pybuffer.shape[0]; /* … */ /* function exit code */ __pyx_r = Py_None; __Pyx_INCREF(Py_None); goto __pyx_L0; __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); { PyObject *__pyx_type, *__pyx_value, *__pyx_tb; __Pyx_PyThreadState_declare __Pyx_PyThreadState_assign __Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_precon.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);} __Pyx_AddTraceback("anuga.utilities.cg_ext.jacobi_precon_c", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = NULL; goto __pyx_L2; __pyx_L0:; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_precon.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __pyx_L2:; __Pyx_XDECREF((PyObject *)__pyx_v_data); __Pyx_XDECREF((PyObject *)__pyx_v_colind); __Pyx_XDECREF((PyObject *)__pyx_v_row_ptr); __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; } /* … */ __pyx_tuple__3 = PyTuple_Pack(7, __pyx_n_s_csr_sparse, __pyx_n_s_precon, __pyx_n_s_M, __pyx_n_s_err, __pyx_n_s_data, __pyx_n_s_colind, __pyx_n_s_row_ptr); if (unlikely(!__pyx_tuple__3)) __PYX_ERR(0, 14, __pyx_L1_error) __Pyx_GOTREF(__pyx_tuple__3); __Pyx_GIVEREF(__pyx_tuple__3); /* … */ __pyx_t_1 = PyCFunction_NewEx(&__pyx_mdef_5anuga_9utilities_6cg_ext_1jacobi_precon_c, NULL, __pyx_n_s_anuga_utilities_cg_ext); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 14, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_jacobi_precon_c, __pyx_t_1) < 0) __PYX_ERR(0, 14, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0; __pyx_codeobj__4 = (PyObject*)__Pyx_PyCode_New(2, 0, 7, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__3, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_anuga_utilities_cg_ext_pyx, __pyx_n_s_jacobi_precon_c, 14, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__4)) __PYX_ERR(0, 14, __pyx_L1_error)
15:
16: cdef int M, err
17: cdef np.ndarray[double, ndim=1, mode="c"] data
18: cdef np.ndarray[long, ndim=1, mode="c"] colind
19: cdef np.ndarray[long, ndim=1, mode="c"] row_ptr
20:
+21: data = csr_sparse.data
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_data); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 21, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 21, __pyx_L1_error) __pyx_t_2 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_t_2, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_4, &__pyx_t_5, &__pyx_t_6); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_v_data, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_6); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_4, __pyx_t_5, __pyx_t_6); } __pyx_t_4 = __pyx_t_5 = __pyx_t_6 = 0; } __pyx_pybuffernd_data.diminfo[0].strides = __pyx_pybuffernd_data.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_data.diminfo[0].shape = __pyx_pybuffernd_data.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 21, __pyx_L1_error) } __pyx_t_2 = 0; __pyx_v_data = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
+22: colind = csr_sparse.colind
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_colind); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 22, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 22, __pyx_L1_error) __pyx_t_7 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_colind.rcbuffer->pybuffer, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_6, &__pyx_t_5, &__pyx_t_4); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_colind.rcbuffer->pybuffer, (PyObject*)__pyx_v_colind, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_6); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_4); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_6, __pyx_t_5, __pyx_t_4); } __pyx_t_6 = __pyx_t_5 = __pyx_t_4 = 0; } __pyx_pybuffernd_colind.diminfo[0].strides = __pyx_pybuffernd_colind.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_colind.diminfo[0].shape = __pyx_pybuffernd_colind.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 22, __pyx_L1_error) } __pyx_t_7 = 0; __pyx_v_colind = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
+23: row_ptr = csr_sparse.row_ptr
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_row_ptr); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 23, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 23, __pyx_L1_error) __pyx_t_8 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer, (PyObject*)__pyx_t_8, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_4, &__pyx_t_5, &__pyx_t_6); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer, (PyObject*)__pyx_v_row_ptr, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_6); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_4, __pyx_t_5, __pyx_t_6); } __pyx_t_4 = __pyx_t_5 = __pyx_t_6 = 0; } __pyx_pybuffernd_row_ptr.diminfo[0].strides = __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_row_ptr.diminfo[0].shape = __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 23, __pyx_L1_error) } __pyx_t_8 = 0; __pyx_v_row_ptr = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
24:
+25: M = row_ptr.shape[0] - 1
__pyx_v_M = ((__pyx_v_row_ptr->dimensions[0]) - 1);
26:
+27: err = _jacobi_precon_c(&data[0], &colind[0], &row_ptr[0], &precon[0], M)
__pyx_t_9 = 0; __pyx_t_10 = 0; __pyx_t_11 = 0; __pyx_t_12 = 0; __pyx_v_err = _jacobi_precon_c((&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_data.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_data.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(long *, __pyx_pybuffernd_colind.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_colind.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(long *, __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_row_ptr.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_precon.rcbuffer->pybuffer.buf, __pyx_t_12, __pyx_pybuffernd_precon.diminfo[0].strides))), __pyx_v_M);
28:
+29: def cg_solve_c(object csr_sparse,\
/* Python wrapper */ static PyObject *__pyx_pw_5anuga_9utilities_6cg_ext_3cg_solve_c(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/ static PyMethodDef __pyx_mdef_5anuga_9utilities_6cg_ext_3cg_solve_c = {"cg_solve_c", (PyCFunction)(void*)(PyCFunctionWithKeywords)__pyx_pw_5anuga_9utilities_6cg_ext_3cg_solve_c, METH_VARARGS|METH_KEYWORDS, 0}; static PyObject *__pyx_pw_5anuga_9utilities_6cg_ext_3cg_solve_c(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) { PyObject *__pyx_v_csr_sparse = 0; PyArrayObject *__pyx_v_x0 = 0; PyArrayObject *__pyx_v_b = 0; int __pyx_v_imax; double __pyx_v_tol; double __pyx_v_a_tol; CYTHON_UNUSED int __pyx_v_bcols; PyObject *__pyx_r = 0; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("cg_solve_c (wrapper)", 0); { static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_csr_sparse,&__pyx_n_s_x0,&__pyx_n_s_b,&__pyx_n_s_imax,&__pyx_n_s_tol,&__pyx_n_s_a_tol,&__pyx_n_s_bcols,0}; PyObject* values[7] = {0,0,0,0,0,0,0}; if (unlikely(__pyx_kwds)) { Py_ssize_t kw_args; const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args); switch (pos_args) { case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6); CYTHON_FALLTHROUGH; case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5); CYTHON_FALLTHROUGH; case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4); CYTHON_FALLTHROUGH; case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3); CYTHON_FALLTHROUGH; case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2); CYTHON_FALLTHROUGH; case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1); CYTHON_FALLTHROUGH; case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0); CYTHON_FALLTHROUGH; case 0: break; default: goto __pyx_L5_argtuple_error; } kw_args = PyDict_Size(__pyx_kwds); switch (pos_args) { case 0: if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_csr_sparse)) != 0)) kw_args--; else goto __pyx_L5_argtuple_error; CYTHON_FALLTHROUGH; case 1: if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_x0)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, 1); __PYX_ERR(0, 29, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 2: if (likely((values[2] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_b)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, 2); __PYX_ERR(0, 29, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 3: if (likely((values[3] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_imax)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, 3); __PYX_ERR(0, 29, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 4: if (likely((values[4] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_tol)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, 4); __PYX_ERR(0, 29, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 5: if (likely((values[5] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_a_tol)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, 5); __PYX_ERR(0, 29, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 6: if (likely((values[6] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_bcols)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, 6); __PYX_ERR(0, 29, __pyx_L3_error) } } if (unlikely(kw_args > 0)) { if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "cg_solve_c") < 0)) __PYX_ERR(0, 29, __pyx_L3_error) } } else if (PyTuple_GET_SIZE(__pyx_args) != 7) { goto __pyx_L5_argtuple_error; } else { values[0] = PyTuple_GET_ITEM(__pyx_args, 0); values[1] = PyTuple_GET_ITEM(__pyx_args, 1); values[2] = PyTuple_GET_ITEM(__pyx_args, 2); values[3] = PyTuple_GET_ITEM(__pyx_args, 3); values[4] = PyTuple_GET_ITEM(__pyx_args, 4); values[5] = PyTuple_GET_ITEM(__pyx_args, 5); values[6] = PyTuple_GET_ITEM(__pyx_args, 6); } __pyx_v_csr_sparse = values[0]; __pyx_v_x0 = ((PyArrayObject *)values[1]); __pyx_v_b = ((PyArrayObject *)values[2]); __pyx_v_imax = __Pyx_PyInt_As_int(values[3]); if (unlikely((__pyx_v_imax == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 32, __pyx_L3_error) __pyx_v_tol = __pyx_PyFloat_AsDouble(values[4]); if (unlikely((__pyx_v_tol == (double)-1) && PyErr_Occurred())) __PYX_ERR(0, 33, __pyx_L3_error) __pyx_v_a_tol = __pyx_PyFloat_AsDouble(values[5]); if (unlikely((__pyx_v_a_tol == (double)-1) && PyErr_Occurred())) __PYX_ERR(0, 34, __pyx_L3_error) __pyx_v_bcols = __Pyx_PyInt_As_int(values[6]); if (unlikely((__pyx_v_bcols == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 35, __pyx_L3_error) } goto __pyx_L4_argument_unpacking_done; __pyx_L5_argtuple_error:; __Pyx_RaiseArgtupleInvalid("cg_solve_c", 1, 7, 7, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 29, __pyx_L3_error) __pyx_L3_error:; __Pyx_AddTraceback("anuga.utilities.cg_ext.cg_solve_c", __pyx_clineno, __pyx_lineno, __pyx_filename); __Pyx_RefNannyFinishContext(); return NULL; __pyx_L4_argument_unpacking_done:; if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_x0), __pyx_ptype_5numpy_ndarray, 0, "x0", 0))) __PYX_ERR(0, 30, __pyx_L1_error) if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_b), __pyx_ptype_5numpy_ndarray, 0, "b", 0))) __PYX_ERR(0, 31, __pyx_L1_error) __pyx_r = __pyx_pf_5anuga_9utilities_6cg_ext_2cg_solve_c(__pyx_self, __pyx_v_csr_sparse, __pyx_v_x0, __pyx_v_b, __pyx_v_imax, __pyx_v_tol, __pyx_v_a_tol, __pyx_v_bcols); int __pyx_lineno = 0; const char *__pyx_filename = NULL; int __pyx_clineno = 0; /* function exit code */ goto __pyx_L0; __pyx_L1_error:; __pyx_r = NULL; __pyx_L0:; __Pyx_RefNannyFinishContext(); return __pyx_r; } static PyObject *__pyx_pf_5anuga_9utilities_6cg_ext_2cg_solve_c(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csr_sparse, PyArrayObject *__pyx_v_x0, PyArrayObject *__pyx_v_b, int __pyx_v_imax, double __pyx_v_tol, double __pyx_v_a_tol, CYTHON_UNUSED int __pyx_v_bcols) { int __pyx_v_M; int __pyx_v_err; PyArrayObject *__pyx_v_data = 0; PyArrayObject *__pyx_v_colind = 0; PyArrayObject *__pyx_v_row_ptr = 0; __Pyx_LocalBuf_ND __pyx_pybuffernd_b; __Pyx_Buffer __pyx_pybuffer_b; __Pyx_LocalBuf_ND __pyx_pybuffernd_colind; __Pyx_Buffer __pyx_pybuffer_colind; __Pyx_LocalBuf_ND __pyx_pybuffernd_data; __Pyx_Buffer __pyx_pybuffer_data; __Pyx_LocalBuf_ND __pyx_pybuffernd_row_ptr; __Pyx_Buffer __pyx_pybuffer_row_ptr; __Pyx_LocalBuf_ND __pyx_pybuffernd_x0; __Pyx_Buffer __pyx_pybuffer_x0; PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("cg_solve_c", 0); __pyx_pybuffer_data.pybuffer.buf = NULL; __pyx_pybuffer_data.refcount = 0; __pyx_pybuffernd_data.data = NULL; __pyx_pybuffernd_data.rcbuffer = &__pyx_pybuffer_data; __pyx_pybuffer_colind.pybuffer.buf = NULL; __pyx_pybuffer_colind.refcount = 0; __pyx_pybuffernd_colind.data = NULL; __pyx_pybuffernd_colind.rcbuffer = &__pyx_pybuffer_colind; __pyx_pybuffer_row_ptr.pybuffer.buf = NULL; __pyx_pybuffer_row_ptr.refcount = 0; __pyx_pybuffernd_row_ptr.data = NULL; __pyx_pybuffernd_row_ptr.rcbuffer = &__pyx_pybuffer_row_ptr; __pyx_pybuffer_x0.pybuffer.buf = NULL; __pyx_pybuffer_x0.refcount = 0; __pyx_pybuffernd_x0.data = NULL; __pyx_pybuffernd_x0.rcbuffer = &__pyx_pybuffer_x0; __pyx_pybuffer_b.pybuffer.buf = NULL; __pyx_pybuffer_b.refcount = 0; __pyx_pybuffernd_b.data = NULL; __pyx_pybuffernd_b.rcbuffer = &__pyx_pybuffer_b; { __Pyx_BufFmt_StackElem __pyx_stack[1]; if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_x0.rcbuffer->pybuffer, (PyObject*)__pyx_v_x0, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 29, __pyx_L1_error) } __pyx_pybuffernd_x0.diminfo[0].strides = __pyx_pybuffernd_x0.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_x0.diminfo[0].shape = __pyx_pybuffernd_x0.rcbuffer->pybuffer.shape[0]; { __Pyx_BufFmt_StackElem __pyx_stack[1]; if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_b.rcbuffer->pybuffer, (PyObject*)__pyx_v_b, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 29, __pyx_L1_error) } __pyx_pybuffernd_b.diminfo[0].strides = __pyx_pybuffernd_b.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_b.diminfo[0].shape = __pyx_pybuffernd_b.rcbuffer->pybuffer.shape[0]; /* … */ /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); { PyObject *__pyx_type, *__pyx_value, *__pyx_tb; __Pyx_PyThreadState_declare __Pyx_PyThreadState_assign __Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_b.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_x0.rcbuffer->pybuffer); __Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);} __Pyx_AddTraceback("anuga.utilities.cg_ext.cg_solve_c", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = NULL; goto __pyx_L2; __pyx_L0:; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_b.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_x0.rcbuffer->pybuffer); __pyx_L2:; __Pyx_XDECREF((PyObject *)__pyx_v_data); __Pyx_XDECREF((PyObject *)__pyx_v_colind); __Pyx_XDECREF((PyObject *)__pyx_v_row_ptr); __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; } /* … */ __pyx_tuple__5 = PyTuple_Pack(12, __pyx_n_s_csr_sparse, __pyx_n_s_x0, __pyx_n_s_b, __pyx_n_s_imax, __pyx_n_s_tol, __pyx_n_s_a_tol, __pyx_n_s_bcols, __pyx_n_s_M, __pyx_n_s_err, __pyx_n_s_data, __pyx_n_s_colind, __pyx_n_s_row_ptr); if (unlikely(!__pyx_tuple__5)) __PYX_ERR(0, 29, __pyx_L1_error) __Pyx_GOTREF(__pyx_tuple__5); __Pyx_GIVEREF(__pyx_tuple__5); /* … */ __pyx_t_1 = PyCFunction_NewEx(&__pyx_mdef_5anuga_9utilities_6cg_ext_3cg_solve_c, NULL, __pyx_n_s_anuga_utilities_cg_ext); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 29, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_cg_solve_c, __pyx_t_1) < 0) __PYX_ERR(0, 29, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0; __pyx_codeobj__6 = (PyObject*)__Pyx_PyCode_New(7, 0, 12, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__5, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_anuga_utilities_cg_ext_pyx, __pyx_n_s_cg_solve_c, 29, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__6)) __PYX_ERR(0, 29, __pyx_L1_error)
30: np.ndarray[double, ndim=1, mode="c"] x0 not None,\
31: np.ndarray[double, ndim=1, mode="c"] b not None,\
32: int imax,\
33: double tol,\
34: double a_tol,\
35: int bcols):
36:
37: cdef int M, err
38: cdef np.ndarray[double, ndim=1, mode="c"] data
39: cdef np.ndarray[long, ndim=1, mode="c"] colind
40: cdef np.ndarray[long, ndim=1, mode="c"] row_ptr
41:
+42: data = csr_sparse.data
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_data); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 42, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 42, __pyx_L1_error) __pyx_t_2 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_t_2, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_4, &__pyx_t_5, &__pyx_t_6); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_v_data, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_6); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_4, __pyx_t_5, __pyx_t_6); } __pyx_t_4 = __pyx_t_5 = __pyx_t_6 = 0; } __pyx_pybuffernd_data.diminfo[0].strides = __pyx_pybuffernd_data.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_data.diminfo[0].shape = __pyx_pybuffernd_data.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 42, __pyx_L1_error) } __pyx_t_2 = 0; __pyx_v_data = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
+43: colind = csr_sparse.colind
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_colind); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 43, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 43, __pyx_L1_error) __pyx_t_7 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_colind.rcbuffer->pybuffer, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_6, &__pyx_t_5, &__pyx_t_4); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_colind.rcbuffer->pybuffer, (PyObject*)__pyx_v_colind, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_6); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_4); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_6, __pyx_t_5, __pyx_t_4); } __pyx_t_6 = __pyx_t_5 = __pyx_t_4 = 0; } __pyx_pybuffernd_colind.diminfo[0].strides = __pyx_pybuffernd_colind.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_colind.diminfo[0].shape = __pyx_pybuffernd_colind.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 43, __pyx_L1_error) } __pyx_t_7 = 0; __pyx_v_colind = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
+44: row_ptr = csr_sparse.row_ptr
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_row_ptr); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 44, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 44, __pyx_L1_error) __pyx_t_8 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer, (PyObject*)__pyx_t_8, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_4, &__pyx_t_5, &__pyx_t_6); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer, (PyObject*)__pyx_v_row_ptr, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_6); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_4, __pyx_t_5, __pyx_t_6); } __pyx_t_4 = __pyx_t_5 = __pyx_t_6 = 0; } __pyx_pybuffernd_row_ptr.diminfo[0].strides = __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_row_ptr.diminfo[0].shape = __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 44, __pyx_L1_error) } __pyx_t_8 = 0; __pyx_v_row_ptr = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
45:
+46: M = row_ptr.shape[0] - 1
__pyx_v_M = ((__pyx_v_row_ptr->dimensions[0]) - 1);
47:
+48: err = _cg_solve_c(&data[0],\
__pyx_t_9 = 0; /* … */ __pyx_v_err = _cg_solve_c((&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_data.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_data.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(long *, __pyx_pybuffernd_colind.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_colind.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(long *, __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_row_ptr.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_b.rcbuffer->pybuffer.buf, __pyx_t_12, __pyx_pybuffernd_b.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_x0.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_x0.diminfo[0].strides))), __pyx_v_imax, __pyx_v_tol, __pyx_v_a_tol, __pyx_v_M);
+49: &colind[0],\
__pyx_t_10 = 0;
+50: &row_ptr[0],\
__pyx_t_11 = 0;
+51: &b[0],\
__pyx_t_12 = 0;
+52: &x0[0],\
__pyx_t_13 = 0;
53: imax,\
54: tol,\
55: a_tol,\
56: M)
57:
+58: return err
__Pyx_XDECREF(__pyx_r); __pyx_t_1 = __Pyx_PyInt_From_int(__pyx_v_err); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 58, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_r = __pyx_t_1; __pyx_t_1 = 0; goto __pyx_L0;
59:
+60: def cg_solve_c_precon(object csr_sparse,\
/* Python wrapper */ static PyObject *__pyx_pw_5anuga_9utilities_6cg_ext_5cg_solve_c_precon(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/ static PyMethodDef __pyx_mdef_5anuga_9utilities_6cg_ext_5cg_solve_c_precon = {"cg_solve_c_precon", (PyCFunction)(void*)(PyCFunctionWithKeywords)__pyx_pw_5anuga_9utilities_6cg_ext_5cg_solve_c_precon, METH_VARARGS|METH_KEYWORDS, 0}; static PyObject *__pyx_pw_5anuga_9utilities_6cg_ext_5cg_solve_c_precon(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) { PyObject *__pyx_v_csr_sparse = 0; PyArrayObject *__pyx_v_x0 = 0; PyArrayObject *__pyx_v_b = 0; int __pyx_v_imax; double __pyx_v_tol; double __pyx_v_a_tol; CYTHON_UNUSED int __pyx_v_bcols; PyArrayObject *__pyx_v_precon = 0; PyObject *__pyx_r = 0; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("cg_solve_c_precon (wrapper)", 0); { static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_csr_sparse,&__pyx_n_s_x0,&__pyx_n_s_b,&__pyx_n_s_imax,&__pyx_n_s_tol,&__pyx_n_s_a_tol,&__pyx_n_s_bcols,&__pyx_n_s_precon,0}; PyObject* values[8] = {0,0,0,0,0,0,0,0}; if (unlikely(__pyx_kwds)) { Py_ssize_t kw_args; const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args); switch (pos_args) { case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7); CYTHON_FALLTHROUGH; case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6); CYTHON_FALLTHROUGH; case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5); CYTHON_FALLTHROUGH; case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4); CYTHON_FALLTHROUGH; case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3); CYTHON_FALLTHROUGH; case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2); CYTHON_FALLTHROUGH; case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1); CYTHON_FALLTHROUGH; case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0); CYTHON_FALLTHROUGH; case 0: break; default: goto __pyx_L5_argtuple_error; } kw_args = PyDict_Size(__pyx_kwds); switch (pos_args) { case 0: if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_csr_sparse)) != 0)) kw_args--; else goto __pyx_L5_argtuple_error; CYTHON_FALLTHROUGH; case 1: if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_x0)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 1); __PYX_ERR(0, 60, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 2: if (likely((values[2] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_b)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 2); __PYX_ERR(0, 60, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 3: if (likely((values[3] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_imax)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 3); __PYX_ERR(0, 60, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 4: if (likely((values[4] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_tol)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 4); __PYX_ERR(0, 60, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 5: if (likely((values[5] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_a_tol)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 5); __PYX_ERR(0, 60, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 6: if (likely((values[6] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_bcols)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 6); __PYX_ERR(0, 60, __pyx_L3_error) } CYTHON_FALLTHROUGH; case 7: if (likely((values[7] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_precon)) != 0)) kw_args--; else { __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, 7); __PYX_ERR(0, 60, __pyx_L3_error) } } if (unlikely(kw_args > 0)) { if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "cg_solve_c_precon") < 0)) __PYX_ERR(0, 60, __pyx_L3_error) } } else if (PyTuple_GET_SIZE(__pyx_args) != 8) { goto __pyx_L5_argtuple_error; } else { values[0] = PyTuple_GET_ITEM(__pyx_args, 0); values[1] = PyTuple_GET_ITEM(__pyx_args, 1); values[2] = PyTuple_GET_ITEM(__pyx_args, 2); values[3] = PyTuple_GET_ITEM(__pyx_args, 3); values[4] = PyTuple_GET_ITEM(__pyx_args, 4); values[5] = PyTuple_GET_ITEM(__pyx_args, 5); values[6] = PyTuple_GET_ITEM(__pyx_args, 6); values[7] = PyTuple_GET_ITEM(__pyx_args, 7); } __pyx_v_csr_sparse = values[0]; __pyx_v_x0 = ((PyArrayObject *)values[1]); __pyx_v_b = ((PyArrayObject *)values[2]); __pyx_v_imax = __Pyx_PyInt_As_int(values[3]); if (unlikely((__pyx_v_imax == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 63, __pyx_L3_error) __pyx_v_tol = __pyx_PyFloat_AsDouble(values[4]); if (unlikely((__pyx_v_tol == (double)-1) && PyErr_Occurred())) __PYX_ERR(0, 64, __pyx_L3_error) __pyx_v_a_tol = __pyx_PyFloat_AsDouble(values[5]); if (unlikely((__pyx_v_a_tol == (double)-1) && PyErr_Occurred())) __PYX_ERR(0, 65, __pyx_L3_error) __pyx_v_bcols = __Pyx_PyInt_As_int(values[6]); if (unlikely((__pyx_v_bcols == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 66, __pyx_L3_error) __pyx_v_precon = ((PyArrayObject *)values[7]); } goto __pyx_L4_argument_unpacking_done; __pyx_L5_argtuple_error:; __Pyx_RaiseArgtupleInvalid("cg_solve_c_precon", 1, 8, 8, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 60, __pyx_L3_error) __pyx_L3_error:; __Pyx_AddTraceback("anuga.utilities.cg_ext.cg_solve_c_precon", __pyx_clineno, __pyx_lineno, __pyx_filename); __Pyx_RefNannyFinishContext(); return NULL; __pyx_L4_argument_unpacking_done:; if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_x0), __pyx_ptype_5numpy_ndarray, 0, "x0", 0))) __PYX_ERR(0, 61, __pyx_L1_error) if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_b), __pyx_ptype_5numpy_ndarray, 0, "b", 0))) __PYX_ERR(0, 62, __pyx_L1_error) if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_precon), __pyx_ptype_5numpy_ndarray, 0, "precon", 0))) __PYX_ERR(0, 67, __pyx_L1_error) __pyx_r = __pyx_pf_5anuga_9utilities_6cg_ext_4cg_solve_c_precon(__pyx_self, __pyx_v_csr_sparse, __pyx_v_x0, __pyx_v_b, __pyx_v_imax, __pyx_v_tol, __pyx_v_a_tol, __pyx_v_bcols, __pyx_v_precon); int __pyx_lineno = 0; const char *__pyx_filename = NULL; int __pyx_clineno = 0; /* function exit code */ goto __pyx_L0; __pyx_L1_error:; __pyx_r = NULL; __pyx_L0:; __Pyx_RefNannyFinishContext(); return __pyx_r; } static PyObject *__pyx_pf_5anuga_9utilities_6cg_ext_4cg_solve_c_precon(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_csr_sparse, PyArrayObject *__pyx_v_x0, PyArrayObject *__pyx_v_b, int __pyx_v_imax, double __pyx_v_tol, double __pyx_v_a_tol, CYTHON_UNUSED int __pyx_v_bcols, PyArrayObject *__pyx_v_precon) { int __pyx_v_M; int __pyx_v_err; PyArrayObject *__pyx_v_data = 0; PyArrayObject *__pyx_v_colind = 0; PyArrayObject *__pyx_v_row_ptr = 0; __Pyx_LocalBuf_ND __pyx_pybuffernd_b; __Pyx_Buffer __pyx_pybuffer_b; __Pyx_LocalBuf_ND __pyx_pybuffernd_colind; __Pyx_Buffer __pyx_pybuffer_colind; __Pyx_LocalBuf_ND __pyx_pybuffernd_data; __Pyx_Buffer __pyx_pybuffer_data; __Pyx_LocalBuf_ND __pyx_pybuffernd_precon; __Pyx_Buffer __pyx_pybuffer_precon; __Pyx_LocalBuf_ND __pyx_pybuffernd_row_ptr; __Pyx_Buffer __pyx_pybuffer_row_ptr; __Pyx_LocalBuf_ND __pyx_pybuffernd_x0; __Pyx_Buffer __pyx_pybuffer_x0; PyObject *__pyx_r = NULL; __Pyx_RefNannyDeclarations __Pyx_RefNannySetupContext("cg_solve_c_precon", 0); __pyx_pybuffer_data.pybuffer.buf = NULL; __pyx_pybuffer_data.refcount = 0; __pyx_pybuffernd_data.data = NULL; __pyx_pybuffernd_data.rcbuffer = &__pyx_pybuffer_data; __pyx_pybuffer_colind.pybuffer.buf = NULL; __pyx_pybuffer_colind.refcount = 0; __pyx_pybuffernd_colind.data = NULL; __pyx_pybuffernd_colind.rcbuffer = &__pyx_pybuffer_colind; __pyx_pybuffer_row_ptr.pybuffer.buf = NULL; __pyx_pybuffer_row_ptr.refcount = 0; __pyx_pybuffernd_row_ptr.data = NULL; __pyx_pybuffernd_row_ptr.rcbuffer = &__pyx_pybuffer_row_ptr; __pyx_pybuffer_x0.pybuffer.buf = NULL; __pyx_pybuffer_x0.refcount = 0; __pyx_pybuffernd_x0.data = NULL; __pyx_pybuffernd_x0.rcbuffer = &__pyx_pybuffer_x0; __pyx_pybuffer_b.pybuffer.buf = NULL; __pyx_pybuffer_b.refcount = 0; __pyx_pybuffernd_b.data = NULL; __pyx_pybuffernd_b.rcbuffer = &__pyx_pybuffer_b; __pyx_pybuffer_precon.pybuffer.buf = NULL; __pyx_pybuffer_precon.refcount = 0; __pyx_pybuffernd_precon.data = NULL; __pyx_pybuffernd_precon.rcbuffer = &__pyx_pybuffer_precon; { __Pyx_BufFmt_StackElem __pyx_stack[1]; if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_x0.rcbuffer->pybuffer, (PyObject*)__pyx_v_x0, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 60, __pyx_L1_error) } __pyx_pybuffernd_x0.diminfo[0].strides = __pyx_pybuffernd_x0.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_x0.diminfo[0].shape = __pyx_pybuffernd_x0.rcbuffer->pybuffer.shape[0]; { __Pyx_BufFmt_StackElem __pyx_stack[1]; if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_b.rcbuffer->pybuffer, (PyObject*)__pyx_v_b, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 60, __pyx_L1_error) } __pyx_pybuffernd_b.diminfo[0].strides = __pyx_pybuffernd_b.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_b.diminfo[0].shape = __pyx_pybuffernd_b.rcbuffer->pybuffer.shape[0]; { __Pyx_BufFmt_StackElem __pyx_stack[1]; if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_precon.rcbuffer->pybuffer, (PyObject*)__pyx_v_precon, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) __PYX_ERR(0, 60, __pyx_L1_error) } __pyx_pybuffernd_precon.diminfo[0].strides = __pyx_pybuffernd_precon.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_precon.diminfo[0].shape = __pyx_pybuffernd_precon.rcbuffer->pybuffer.shape[0]; /* … */ /* function exit code */ __pyx_L1_error:; __Pyx_XDECREF(__pyx_t_1); { PyObject *__pyx_type, *__pyx_value, *__pyx_tb; __Pyx_PyThreadState_declare __Pyx_PyThreadState_assign __Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_b.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_precon.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_x0.rcbuffer->pybuffer); __Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);} __Pyx_AddTraceback("anuga.utilities.cg_ext.cg_solve_c_precon", __pyx_clineno, __pyx_lineno, __pyx_filename); __pyx_r = NULL; goto __pyx_L2; __pyx_L0:; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_b.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_precon.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_x0.rcbuffer->pybuffer); __pyx_L2:; __Pyx_XDECREF((PyObject *)__pyx_v_data); __Pyx_XDECREF((PyObject *)__pyx_v_colind); __Pyx_XDECREF((PyObject *)__pyx_v_row_ptr); __Pyx_XGIVEREF(__pyx_r); __Pyx_RefNannyFinishContext(); return __pyx_r; } /* … */ __pyx_tuple__7 = PyTuple_Pack(13, __pyx_n_s_csr_sparse, __pyx_n_s_x0, __pyx_n_s_b, __pyx_n_s_imax, __pyx_n_s_tol, __pyx_n_s_a_tol, __pyx_n_s_bcols, __pyx_n_s_precon, __pyx_n_s_M, __pyx_n_s_err, __pyx_n_s_data, __pyx_n_s_colind, __pyx_n_s_row_ptr); if (unlikely(!__pyx_tuple__7)) __PYX_ERR(0, 60, __pyx_L1_error) __Pyx_GOTREF(__pyx_tuple__7); __Pyx_GIVEREF(__pyx_tuple__7); /* … */ __pyx_t_1 = PyCFunction_NewEx(&__pyx_mdef_5anuga_9utilities_6cg_ext_5cg_solve_c_precon, NULL, __pyx_n_s_anuga_utilities_cg_ext); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 60, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (PyDict_SetItem(__pyx_d, __pyx_n_s_cg_solve_c_precon, __pyx_t_1) < 0) __PYX_ERR(0, 60, __pyx_L1_error) __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
61: np.ndarray[double, ndim=1, mode="c"] x0 not None,\
62: np.ndarray[double, ndim=1, mode="c"] b not None,\
63: int imax,\
64: double tol,\
65: double a_tol,\
66: int bcols,\
67: np.ndarray[double, ndim=1, mode="c"] precon not None):
68:
69: cdef int M, err
70: cdef np.ndarray[double, ndim=1, mode="c"] data
71: cdef np.ndarray[long, ndim=1, mode="c"] colind
72: cdef np.ndarray[long, ndim=1, mode="c"] row_ptr
73:
+74: data = csr_sparse.data
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_data); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 74, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 74, __pyx_L1_error) __pyx_t_2 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_t_2, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_4, &__pyx_t_5, &__pyx_t_6); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_v_data, &__Pyx_TypeInfo_double, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_6); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_4, __pyx_t_5, __pyx_t_6); } __pyx_t_4 = __pyx_t_5 = __pyx_t_6 = 0; } __pyx_pybuffernd_data.diminfo[0].strides = __pyx_pybuffernd_data.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_data.diminfo[0].shape = __pyx_pybuffernd_data.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 74, __pyx_L1_error) } __pyx_t_2 = 0; __pyx_v_data = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
+75: colind = csr_sparse.colind
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_colind); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 75, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 75, __pyx_L1_error) __pyx_t_7 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_colind.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_colind.rcbuffer->pybuffer, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_6, &__pyx_t_5, &__pyx_t_4); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_colind.rcbuffer->pybuffer, (PyObject*)__pyx_v_colind, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_6); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_4); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_6, __pyx_t_5, __pyx_t_4); } __pyx_t_6 = __pyx_t_5 = __pyx_t_4 = 0; } __pyx_pybuffernd_colind.diminfo[0].strides = __pyx_pybuffernd_colind.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_colind.diminfo[0].shape = __pyx_pybuffernd_colind.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 75, __pyx_L1_error) } __pyx_t_7 = 0; __pyx_v_colind = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
+76: row_ptr = csr_sparse.row_ptr
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_csr_sparse, __pyx_n_s_row_ptr); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 76, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 76, __pyx_L1_error) __pyx_t_8 = ((PyArrayObject *)__pyx_t_1); { __Pyx_BufFmt_StackElem __pyx_stack[1]; __Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer); __pyx_t_3 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer, (PyObject*)__pyx_t_8, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack); if (unlikely(__pyx_t_3 < 0)) { PyErr_Fetch(&__pyx_t_4, &__pyx_t_5, &__pyx_t_6); if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_row_ptr.rcbuffer->pybuffer, (PyObject*)__pyx_v_row_ptr, &__Pyx_TypeInfo_long, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) { Py_XDECREF(__pyx_t_4); Py_XDECREF(__pyx_t_5); Py_XDECREF(__pyx_t_6); __Pyx_RaiseBufferFallbackError(); } else { PyErr_Restore(__pyx_t_4, __pyx_t_5, __pyx_t_6); } __pyx_t_4 = __pyx_t_5 = __pyx_t_6 = 0; } __pyx_pybuffernd_row_ptr.diminfo[0].strides = __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_row_ptr.diminfo[0].shape = __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.shape[0]; if (unlikely(__pyx_t_3 < 0)) __PYX_ERR(0, 76, __pyx_L1_error) } __pyx_t_8 = 0; __pyx_v_row_ptr = ((PyArrayObject *)__pyx_t_1); __pyx_t_1 = 0;
77:
+78: M = row_ptr.shape[0] - 1
__pyx_v_M = ((__pyx_v_row_ptr->dimensions[0]) - 1);
79:
+80: err = _cg_solve_c_precon(&data[0],\
__pyx_t_9 = 0; /* … */ __pyx_v_err = _cg_solve_c_precon((&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_data.rcbuffer->pybuffer.buf, __pyx_t_9, __pyx_pybuffernd_data.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(long *, __pyx_pybuffernd_colind.rcbuffer->pybuffer.buf, __pyx_t_10, __pyx_pybuffernd_colind.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(long *, __pyx_pybuffernd_row_ptr.rcbuffer->pybuffer.buf, __pyx_t_11, __pyx_pybuffernd_row_ptr.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_b.rcbuffer->pybuffer.buf, __pyx_t_12, __pyx_pybuffernd_b.diminfo[0].strides))), (&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_x0.rcbuffer->pybuffer.buf, __pyx_t_13, __pyx_pybuffernd_x0.diminfo[0].strides))), __pyx_v_imax, __pyx_v_tol, __pyx_v_a_tol, __pyx_v_M, (&(*__Pyx_BufPtrCContig1d(double *, __pyx_pybuffernd_precon.rcbuffer->pybuffer.buf, __pyx_t_14, __pyx_pybuffernd_precon.diminfo[0].strides))));
+81: &colind[0],\
__pyx_t_10 = 0;
+82: &row_ptr[0],\
__pyx_t_11 = 0;
+83: &b[0],\
__pyx_t_12 = 0;
+84: &x0[0],\
__pyx_t_13 = 0;
85: imax,\
86: tol,\
87: a_tol,\
88: M,\
+89: &precon[0])
__pyx_t_14 = 0;
90:
+91: return err
__Pyx_XDECREF(__pyx_r); __pyx_t_1 = __Pyx_PyInt_From_int(__pyx_v_err); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 91, __pyx_L1_error) __Pyx_GOTREF(__pyx_t_1); __pyx_r = __pyx_t_1; __pyx_t_1 = 0; goto __pyx_L0;