o Eba@sdZddlmZddlmZddlZejeje Z ddddd d d d Z d dddddd dddddddddZ ddZ dZddddddddddd Zd d!Zd"d#Zd$Zd%d&Zd'Zd(d)Zd*Zd+d,Zd-Zd.d/Zd0Zd1d2Zd3Zd4Zd5d6Zd7Zd8d9Zd:Z d;d<Z!d=Z"d>d?d@d>d>d>d>d>d>dA Z#dBdBdCZ$dDd>iZ%ee&e$e%e%e$e$dEZ'dFdGZ(dHdIZ)dJdKZ*dLdMZ+dNdOZ,dPZ-dQdRZ.dSZ/dTdUZ0dVZ1dWZ2dXZ3dYZ4dZd[Z5d\d]Z6d^d_Z7d`daZ8 b c d e f g h idmdjdkZ9e:dlkre9dSdS)nz Code generator script to make the Cython BLAS and LAPACK wrappers from the files "cython_blas_signatures.txt" and "cython_lapack_signatures.txt" which contain the signatures for all the BLAS/LAPACK routines that should be included in the wrappers. ) defaultdict) itemgetterNZintegercomplexzdouble precisionrealz complex*16 characterZlogical)intcdszcharbintr npy_complex64Zdoublefloatnpy_complex128r _cselect1 _cselect2 _dselect2 _dselect3 _sselect2 _sselect3 _zselect1 _zselect2)rrr r r r r cselect1cselect2dselect2dselect3sselect2sselect3zselect1zselect2cCstdd|dDS)NcSsg|]}|dqSz *split.0argr'(z'arg_names_and_types.., )zipr#)argsr'r'r(arg_names_and_types'r.z cdef extern from "{header_name}": void _fortran_{name} "F_FUNC({name}wrp, {upname}WRP)"({ret_type} *out, {fort_args}) nogil cdef {ret_type} {name}({args}) nogil: cdef {ret_type} out _fortran_{name}(&out, {argnames}) return out ) rr rrrrrrrr cCs|dvr d|SdS)N) rrrrrrrrrrz<{0}*>format)r&r'r'r( arg_casts;s r3c st|\}}|vr'fdd|D}dd|D}dddt||D}dd|D}dddt||D}ddt||D}d|}t}|d d }tj|||||||d S) Ncs g|] }|kr |ndqS)_r'r%nret_typer'r(r)J z!pyx_decl_func..cS g|] }|dvr |n|dqS)lambdainr4r'r5r'r'r(r)Ksr+cSg|] \}}d||gqSr!joinr%r6tr'r'r(r)McSg|]}t||qSr' npy_typesgetr%rBr'r'r(r)OcSr>r!r?rAr'r'r(r)PrCcSg|] \}}t||qSr'r3rAr'r'r(r)Rr<lambda_)nameupnamer- fort_argsr8 c_ret_typeargnames header_name)r.r@r,c_typesreplacepyx_func_templater2upper)rNr8r-rSargtypesrRrPrQr'r7r( pyx_decl_funcCs,     rYzcdef extern from "{header_name}": void _fortran_{name} "F_FUNC({name},{upname})"({fort_args}) nogil cdef void {name}({args}) nogil: _fortran_{name}({argnames}) cCst|\}}dd|D}dd|D}dddt||D}ddt||D}d|}|ddd d }tj||||||d S) NcSrDr'rErHr'r'r(r)erIz pyx_decl_sub..cSr:r;r'r5r'r'r(r)fr9r+cSr>r!r?rAr'r'r(r)grCcSrJr'rKrAr'r'r(r)irLz*lambda,z *lambda_,*in,*in_,)rNrOr-rPrRrS)r.r@r,rUpyx_sub_templater2rW)rNr-rSrXrRrPr'r'r( pyx_decl_subcs    r]a# cython: boundscheck = False # cython: wraparound = False # cython: cdivision = True """ BLAS Functions for Cython ========================= Usable from Cython via:: cimport scipy.linalg.cython_blas These wrappers do not check for alignment of arrays. Alignment should be checked before these wrappers are used. Raw function pointers (Fortran-style pointer arguments): - {} """ # Within SciPy, these wrappers can be used via relative or absolute cimport. # Examples: # from ..linalg cimport cython_blas # from scipy.linalg cimport cython_blas # cimport scipy.linalg.cython_blas as cython_blas # cimport ..linalg.cython_blas as cython_blas # Within SciPy, if BLAS functions are needed in C/C++/Fortran, # these wrappers should not be used. # The original libraries should be linked directly. cdef extern from "fortran_defs.h": pass from numpy cimport npy_complex64, npy_complex128 cCdd|D}td|S)NcSg|]}|dqSrr'r%sigr'r'r(r)z*make_blas_pyx_preamble.. - )blas_pyx_preambler2r@all_sigsnamesr'r'r(make_blas_pyx_preambleria"""" LAPACK functions for Cython =========================== Usable from Cython via:: cimport scipy.linalg.cython_lapack This module provides Cython-level wrappers for all primary routines included in LAPACK 3.4.0 except for ``zcgesv`` since its interface is not consistent from LAPACK 3.4.0 to 3.6.0. It also provides some of the fixed-api auxiliary routines. These wrappers do not check for alignment of arrays. Alignment should be checked before these wrappers are used. Raw function pointers (Fortran-style pointer arguments): - {} """ # Within SciPy, these wrappers can be used via relative or absolute cimport. # Examples: # from ..linalg cimport cython_lapack # from scipy.linalg cimport cython_lapack # cimport scipy.linalg.cython_lapack as cython_lapack # cimport ..linalg.cython_lapack as cython_lapack # Within SciPy, if LAPACK functions are needed in C/C++/Fortran, # these wrappers should not be used. # The original libraries should be linked directly. cdef extern from "fortran_defs.h": pass from numpy cimport npy_complex64, npy_complex128 cdef extern from "_lapack_subroutines.h": # Function pointer type declarations for # gees and gges families of functions. ctypedef bint _cselect1(npy_complex64*) ctypedef bint _cselect2(npy_complex64*, npy_complex64*) ctypedef bint _dselect2(d*, d*) ctypedef bint _dselect3(d*, d*, d*) ctypedef bint _sselect2(s*, s*) ctypedef bint _sselect3(s*, s*, s*) ctypedef bint _zselect1(npy_complex128*) ctypedef bint _zselect2(npy_complex128*, npy_complex128*) cCr^)NcSr_r`r'rar'r'r(r)rcz,make_lapack_pyx_preamble..rd)lapack_pyx_preambler2r@rfr'r'r(make_lapack_pyx_preamblerjrla # Python-accessible wrappers for testing: cdef inline bint _is_contiguous(double[:,:] a, int axis) nogil: return (a.strides[axis] == sizeof(a[0,0]) or a.shape[axis] == 1) cpdef float complex _test_cdotc(float complex[:] cx, float complex[:] cy) nogil: cdef: int n = cx.shape[0] int incx = cx.strides[0] // sizeof(cx[0]) int incy = cy.strides[0] // sizeof(cy[0]) return cdotc(&n, &cx[0], &incx, &cy[0], &incy) cpdef float complex _test_cdotu(float complex[:] cx, float complex[:] cy) nogil: cdef: int n = cx.shape[0] int incx = cx.strides[0] // sizeof(cx[0]) int incy = cy.strides[0] // sizeof(cy[0]) return cdotu(&n, &cx[0], &incx, &cy[0], &incy) cpdef double _test_dasum(double[:] dx) nogil: cdef: int n = dx.shape[0] int incx = dx.strides[0] // sizeof(dx[0]) return dasum(&n, &dx[0], &incx) cpdef double _test_ddot(double[:] dx, double[:] dy) nogil: cdef: int n = dx.shape[0] int incx = dx.strides[0] // sizeof(dx[0]) int incy = dy.strides[0] // sizeof(dy[0]) return ddot(&n, &dx[0], &incx, &dy[0], &incy) cpdef int _test_dgemm(double alpha, double[:,:] a, double[:,:] b, double beta, double[:,:] c) nogil except -1: cdef: char *transa char *transb int m, n, k, lda, ldb, ldc double *a0=&a[0,0] double *b0=&b[0,0] double *c0=&c[0,0] # In the case that c is C contiguous, swap a and b and # swap whether or not each of them is transposed. # This can be done because a.dot(b) = b.T.dot(a.T).T. if _is_contiguous(c, 1): if _is_contiguous(a, 1): transb = 'n' ldb = (&a[1,0]) - a0 if a.shape[0] > 1 else 1 elif _is_contiguous(a, 0): transb = 't' ldb = (&a[0,1]) - a0 if a.shape[1] > 1 else 1 else: with gil: raise ValueError("Input 'a' is neither C nor Fortran contiguous.") if _is_contiguous(b, 1): transa = 'n' lda = (&b[1,0]) - b0 if b.shape[0] > 1 else 1 elif _is_contiguous(b, 0): transa = 't' lda = (&b[0,1]) - b0 if b.shape[1] > 1 else 1 else: with gil: raise ValueError("Input 'b' is neither C nor Fortran contiguous.") k = b.shape[0] if k != a.shape[1]: with gil: raise ValueError("Shape mismatch in input arrays.") m = b.shape[1] n = a.shape[0] if n != c.shape[0] or m != c.shape[1]: with gil: raise ValueError("Output array does not have the correct shape.") ldc = (&c[1,0]) - c0 if c.shape[0] > 1 else 1 dgemm(transa, transb, &m, &n, &k, &alpha, b0, &lda, a0, &ldb, &beta, c0, &ldc) elif _is_contiguous(c, 0): if _is_contiguous(a, 1): transa = 't' lda = (&a[1,0]) - a0 if a.shape[0] > 1 else 1 elif _is_contiguous(a, 0): transa = 'n' lda = (&a[0,1]) - a0 if a.shape[1] > 1 else 1 else: with gil: raise ValueError("Input 'a' is neither C nor Fortran contiguous.") if _is_contiguous(b, 1): transb = 't' ldb = (&b[1,0]) - b0 if b.shape[0] > 1 else 1 elif _is_contiguous(b, 0): transb = 'n' ldb = (&b[0,1]) - b0 if b.shape[1] > 1 else 1 else: with gil: raise ValueError("Input 'b' is neither C nor Fortran contiguous.") m = a.shape[0] k = a.shape[1] if k != b.shape[0]: with gil: raise ValueError("Shape mismatch in input arrays.") n = b.shape[1] if m != c.shape[0] or n != c.shape[1]: with gil: raise ValueError("Output array does not have the correct shape.") ldc = (&c[0,1]) - c0 if c.shape[1] > 1 else 1 dgemm(transa, transb, &m, &n, &k, &alpha, a0, &lda, b0, &ldb, &beta, c0, &ldc) else: with gil: raise ValueError("Input 'c' is neither C nor Fortran contiguous.") return 0 cpdef double _test_dnrm2(double[:] x) nogil: cdef: int n = x.shape[0] int incx = x.strides[0] // sizeof(x[0]) return dnrm2(&n, &x[0], &incx) cpdef double _test_dzasum(double complex[:] zx) nogil: cdef: int n = zx.shape[0] int incx = zx.strides[0] // sizeof(zx[0]) return dzasum(&n, &zx[0], &incx) cpdef double _test_dznrm2(double complex[:] x) nogil: cdef: int n = x.shape[0] int incx = x.strides[0] // sizeof(x[0]) return dznrm2(&n, &x[0], &incx) cpdef int _test_icamax(float complex[:] cx) nogil: cdef: int n = cx.shape[0] int incx = cx.strides[0] // sizeof(cx[0]) return icamax(&n, &cx[0], &incx) cpdef int _test_idamax(double[:] dx) nogil: cdef: int n = dx.shape[0] int incx = dx.strides[0] // sizeof(dx[0]) return idamax(&n, &dx[0], &incx) cpdef int _test_isamax(float[:] sx) nogil: cdef: int n = sx.shape[0] int incx = sx.strides[0] // sizeof(sx[0]) return isamax(&n, &sx[0], &incx) cpdef int _test_izamax(double complex[:] zx) nogil: cdef: int n = zx.shape[0] int incx = zx.strides[0] // sizeof(zx[0]) return izamax(&n, &zx[0], &incx) cpdef float _test_sasum(float[:] sx) nogil: cdef: int n = sx.shape[0] int incx = sx.shape[0] // sizeof(sx[0]) return sasum(&n, &sx[0], &incx) cpdef float _test_scasum(float complex[:] cx) nogil: cdef: int n = cx.shape[0] int incx = cx.strides[0] // sizeof(cx[0]) return scasum(&n, &cx[0], &incx) cpdef float _test_scnrm2(float complex[:] x) nogil: cdef: int n = x.shape[0] int incx = x.strides[0] // sizeof(x[0]) return scnrm2(&n, &x[0], &incx) cpdef float _test_sdot(float[:] sx, float[:] sy) nogil: cdef: int n = sx.shape[0] int incx = sx.strides[0] // sizeof(sx[0]) int incy = sy.strides[0] // sizeof(sy[0]) return sdot(&n, &sx[0], &incx, &sy[0], &incy) cpdef float _test_snrm2(float[:] x) nogil: cdef: int n = x.shape[0] int incx = x.shape[0] // sizeof(x[0]) return snrm2(&n, &x[0], &incx) cpdef double complex _test_zdotc(double complex[:] zx, double complex[:] zy) nogil: cdef: int n = zx.shape[0] int incx = zx.strides[0] // sizeof(zx[0]) int incy = zy.strides[0] // sizeof(zy[0]) return zdotc(&n, &zx[0], &incx, &zy[0], &incy) cpdef double complex _test_zdotu(double complex[:] zx, double complex[:] zy) nogil: cdef: int n = zx.shape[0] int incx = zx.strides[0] // sizeof(zx[0]) int incy = zy.strides[0] // sizeof(zy[0]) return zdotu(&n, &zx[0], &incx, &zy[0], &incy) csHdfdd|D}ddfdd|D}t|||tS)N c3|] }t|fVqdSNrYr%r rSr'r( z$generate_blas_pyx..c3(|]}t|dddfVqdSNr]rqrrr'r(rs )r@riblas_py_wrappers) func_sigssub_sigsrgrSfuncssubsr'rrr(generate_blas_pyxs  ra # Python accessible wrappers for testing: def _test_dlamch(cmach): # This conversion is necessary to handle Python 3 strings. cmach_bytes = bytes(cmach) # Now that it is a bytes representation, a non-temporary variable # must be passed as a part of the function call. cdef char* cmach_char = cmach_bytes return dlamch(cmach_char) def _test_slamch(cmach): # This conversion is necessary to handle Python 3 strings. cmach_bytes = bytes(cmach) # Now that it is a bytes representation, a non-temporary variable # must be passed as a part of the function call. cdef char* cmach_char = cmach_bytes return slamch(cmach_char) csLdfdd|D}ddfdd|D}t|}|||tS)Nrmc3rnrorprqrrr'r(rsrtz&generate_lapack_pyx..c3rurvrxrqrrr'r(rsry)r@rllapack_py_wrappers)r{r|rgrSr}r~preambler'rrr(generate_lapack_pyxs  rzCctypedef {ret_type} {name}_t({args}) nogil cdef {name}_t *{name}_f z%cdef {ret_type} {name}({args}) nogil cCs$|dddd}tj|||dS)Nr<rMrZr[rNr8r-)rU pxd_templater2rr'r'r(pxd_declsra# Within scipy, these wrappers can be used via relative or absolute cimport. # Examples: # from ..linalg cimport cython_blas # from scipy.linalg cimport cython_blas # cimport scipy.linalg.cython_blas as cython_blas # cimport ..linalg.cython_blas as cython_blas # Within SciPy, if BLAS functions are needed in C/C++/Fortran, # these wrappers should not be used. # The original libraries should be linked directly. ctypedef float s ctypedef double d ctypedef float complex c ctypedef double complex z cCsddd|D}t|S)Nrmcs|]}t|VqdSrorrar'r'r(rsz$generate_blas_pxd..)r@blas_pxd_preamble)rgbodyr'r'r(generate_blas_pxdsraM# Within SciPy, these wrappers can be used via relative or absolute cimport. # Examples: # from ..linalg cimport cython_lapack # from scipy.linalg cimport cython_lapack # cimport scipy.linalg.cython_lapack as cython_lapack # cimport ..linalg.cython_lapack as cython_lapack # Within SciPy, if LAPACK functions are needed in C/C++/Fortran, # these wrappers should not be used. # The original libraries should be linked directly. ctypedef float s ctypedef double d ctypedef float complex c ctypedef double complex z # Function pointer type declarations for # gees and gges families of functions. ctypedef bint cselect1(c*) ctypedef bint cselect2(c*, c*) ctypedef bint dselect2(d*, d*) ctypedef bint dselect3(d*, d*, d*) ctypedef bint sselect2(s*, s*) ctypedef bint sselect3(s*, s*, s*) ctypedef bint zselect1(z*) ctypedef bint zselect2(z*, z*) cCstddd|DS)Nrmcsrrorrar'r'r(rs rz&generate_lapack_pxd..)lapack_pxd_preambler@)rgr'r'r(generate_lapack_pxd r/rz subroutine {name}wrp( + ret, + {argnames} + ) external {wrapper} {ret_type} {wrapper} {ret_type} ret {argdecls} ret = {wrapper}( + {argnames} + ) end z(*)z(ldab,*)z(lda,*)) ZworkabaZdlr ZduZapeZlldr0)xyr)ZladivZlanhfZlansfZlapy2Zlapy3cCsTd|vr|St|dd}d|vsd|vr ||d}||S||d}||S)NZincrrz(n)r0) special_casesrG)rNfuncnameZspecialsuffixr'r'r(process_fortran_name*s  rcCsgd}||vr d|S|S)N)ZcdotcZcdotuZzdotcZzdotuZcladivZzladivwr')rNZincludedr'r'r( called_name5srcsbt}t|\}}d|}fdd|D}dddt||D}tj|||t|dS)Nz , + csg|]}t|qSr')rr5rNr'r(r)Ar*z+fort_subroutine_wrapper..z css$|] \}}dt||VqdS)z{0} {1}N)r2 fortran_typesrAr'r'r(rsBsz*fort_subroutine_wrapper..)rNwrapperrRargdeclsr8)rr.r@r,fortran_templater2r)rNr8r-rtypesrhrRrr'rr(fort_subroutine_wrapper<s   rcCsddd|DS)Nrmcsrro)rrar'r'r(rsJrz#generate_fortran..r?)r{r'r'r(generate_fortranIsrcCs4t|\}}dd|D}dddt||DS)NcSsg|]}t|qSr')rTr$r'r'r(r)Orczmake_c_args..r+css |] \}}d||VqdS)z{0} *{1}Nr1)r%rBr6r'r'r(rsPszmake_c_args..)r.r@r,)r-rrhr'r'r( make_c_argsMs rzAvoid F_FUNC({name}wrp, {upname}WRP)({return_type} *ret, {args}); cCs&t|}t|}tj||||dS)N)rNrO return_typer-)rrTc_func_templater2rWrNrr-r'r'r( c_func_declWs  rz&void F_FUNC({name},{upname})({args}); cCst|}tj|||dS)N)rNrOr-)rc_sub_templater2rWrr'r'r( c_sub_declasrz#ifndef SCIPY_LINALG_{lib}_FORTRAN_WRAPPERS_H #define SCIPY_LINALG_{lib}_FORTRAN_WRAPPERS_H #include "fortran_defs.h" #include "numpy/arrayobject.h" a typedef int (*_cselect1)(npy_complex64*); typedef int (*_cselect2)(npy_complex64*, npy_complex64*); typedef int (*_dselect2)(double*, double*); typedef int (*_dselect3)(double*, double*, double*); typedef int (*_sselect2)(float*, float*); typedef int (*_sselect3)(float*, float*, float*); typedef int (*_zselect1)(npy_complex128*); typedef int (*_zselect2)(npy_complex128*, npy_complex128*); z) #ifdef __cplusplus extern "C" { #endif z$ #ifdef __cplusplus } #endif #endif cCsfddd|D}dddd|D}|dkr#tj|dt}ntj|d}d|t||tgS)Nr0csrro)rrar'r'r(rsrz$generate_c_header..rmcsrro)rrar'r'r(rsrLAPACK)lib)r@ c_preambler2 lapack_decls cpp_guardc_end)r{r|rgZlib_namer}r~rr'r'r(generate_c_headers  rcCs.|ddd\}}|d\}}|||fS)N( r")rbZ name_and_typer-r8rNr'r'r(split_signatures rcCsTddttj|D}dd|D}dd|D}tt||tdd}|||fS)NcSsg|] }|r|ds|qS)#) startswithr%liner'r'r(r)s  z filter_lines..cSs&g|]}|dddkrt|qSrrZvoidr#rrr'r'r(r)cSs&g|]}|dddkrt|qSrrrr'r'r(r)rr)key)mapstrstriplistsortedr)linesr{r|rgr'r'r( filter_liness  rcs$ddlmtfdd|DS)Nr)newerc3s0|]}D]}tj|o||VqqdSro)ospathexists)r%dstsrcr src_filesr'r(rss  zall_newer..)Zdistutils.dep_utilrall)r dst_filesr'rr( all_newers rcython_blas_signatures.txtcython_lapack_signatures.txt cython_blas cython_lapack_blas_subroutine_wrappers.f_lapack_subroutine_wrappers.f_blas_subroutines.h_lapack_subroutines.hcCstjt||f}|d|d|||d|d||f} ttt|| r+tddSddg} ddd| Dd } dd d| Dd } dd d| Dd } t |d  }| }Wdn1sgwYt |}t ||f}t |dd }| | | |Wdn1swYt|d}t |dd }| | | |Wdn1swYt|d}t |d }| | | |Wdn1swYt|d}t |d }| | | |Wdn 1swYt |d  }| }Wdn 1swYt |}t||f}t |dd }| | | |Wdn 1sLwYt|d}t |dd }| | | |Wdn 1stwYt|d}t |d }| | | |Wdn 1swYt|d}t |d }| | | |WddS1swYdS)Nz.pyxz.pxdz3scipy/linalg/_generate_pyx.py: all files up-to-datez-This file was generated by _generate_pyx.py. z Do not edit this file directly. r0cSsg|] }d|dqS)z/* z */ )rstriprr'r'r(r)szmake_all..rmcSg|]}d|qS)z# r'rr'r'r(r)rccSr)zc r'rr'r'r(r)rcrrrwr)ZBLAS)r)rrabspath__file__chdirBASE_DIRrprintr@open readlinesrrwriterrrrr)Zblas_signature_fileZlapack_signature_fileZ blas_nameZ lapack_nameZblas_fortran_nameZlapack_fortran_nameZblas_header_nameZlapack_header_namerrZcommentsZccommentZ pyxcommentZfcommentfZ blas_sigsZblas_pyxZblas_pxdZ blas_fortranZ blas_c_headerZ lapack_sigsZ lapack_pyxZ lapack_pxdZlapack_fortranZlapack_c_headerr'r'r(make_alls                                 $r__main__)rrrrrrrr);__doc__ collectionsroperatorrrrrdirnamerrrrTr.rVrFr3rYr\r]rerirkrlrzrrrrrrrrrrZdimsZxy_specialized_dimsZa_specialized_dimsdictrrrrrrrrrrrrrrrrrrr__name__r'r'r'r(s   )6K        I