B 0da@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`daZ8dmdjdkZ9e:dlkre9dS)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) itemgetterNintegercomplexzdouble precisionrealz complex*16 characterZlogical)intcdszcharbintr npy_complex64doublefloatnpy_complex128r _cselect1 _cselect2 _dselect2 _dselect3 _sselect2 _sselect3 _zselect1 _zselect2)rr r r r r rcselect1cselect2dselect2dselect3sselect2sselect3zselect1zselect2cCstdd|dDS)NcSsg|]}|dqS)z *)split).0argr&L/var/www/html/venv/lib/python3.7/site-packages/scipy/linalg/_generate_pyx.py (sz'arg_names_and_types..z, )zipr#)argsr&r&r'arg_names_and_types'sr+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 ) r r rrrrrr r!r"cCs|dkrd|SdS)N) rrrrrrrrrrz<{0}*>)format)r%r&r&r' arg_casts;s r.c st|\}}|krNfdd|D}dd|D}dddt||D}dd|D}dddt||D}ddt||D}d|}t}|d d }tj|||||||d S) Ncs g|]}|kr|ndqS)_r&)r$n)ret_typer&r'r(Jsz!pyx_decl_func..cSs g|]}|dkr|n|dqS))lambdainr/r&)r$r0r&r&r'r(Ksz, cSsg|]\}}d||gqS)z *)join)r$r0tr&r&r'r(MscSsg|]}t||qSr&) npy_typesget)r$r5r&r&r'r(OscSsg|]\}}d||gqS)z *)r4)r$r0r5r&r&r'r(PscSsg|]\}}t||qSr&)r.)r$r0r5r&r&r'r(Rsr2lambda_)nameupnamer* fort_argsr1 c_ret_typeargnames header_name)r+r4r)c_typesreplacepyx_func_templater-upper)r9r1r*r>argtypesr=r;r<r&)r1r' pyx_decl_funcCs$     rDzcdef 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) NcSsg|]}t||qSr&)r6r7)r$r5r&r&r'r(esz pyx_decl_sub..cSs g|]}|dkr|n|dqS))r2r3r/r&)r$r0r&r&r'r(fsz, cSsg|]\}}d||gqS)z *)r4)r$r0r5r&r&r'r(gscSsg|]\}}t||qSr&)r.)r$r0r5r&r&r'r(isz*lambda,z *lambda_,z*in,z*in_,)r9r:r*r;r=r>)r+r4r)r@pyx_sub_templater-rB)r9r*r>rCr=r;r&r&r' pyx_decl_subcs    rFa# 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 cCsdd|D}td|S)NcSsg|] }|dqS)rr&)r$sigr&r&r'r(sz*make_blas_pyx_preamble..z - )blas_pyx_preambler-r4)all_sigsnamesr&r&r'make_blas_pyx_preamblesrKa"""" 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*) cCsdd|D}td|S)NcSsg|] }|dqS)rr&)r$rGr&r&r'r(sz,make_lapack_pyx_preamble..z - )lapack_pyx_preambler-r4)rIrJr&r&r'make_lapack_pyx_preamblesrMa # 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 c3s|]}t|fVqdS)N)rD)r$r )r>r&r' sz$generate_blas_pyx..c3s&|]}t|dddfVqdS)N)rF)r$r )r>r&r'rOs)r4rKblas_py_wrappers) func_sigssub_sigsrIr>funcssubsr&)r>r'generate_blas_pyxs rVa # 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)NrNc3s|]}t|fVqdS)N)rD)r$r )r>r&r'rOsz&generate_lapack_pyx..c3s&|]}t|dddfVqdS)NrP)rF)r$r )r>r&r'rOs)r4rMlapack_py_wrappers)rRrSrIr>rTrUpreambler&)r>r'generate_lapack_pyxs  rYzCctypedef {ret_type} {name}_t({args}) nogil cdef {name}_t *{name}_f z%cdef {ret_type} {name}({args}) nogil cCs$|dddd}tj|||dS)Nr2r8z*in,z*in_,)r9r1r*)r@ pxd_templater-)r9r1r*r&r&r'pxd_declsr[a# 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)NrNcss|]}t|VqdS)N)r[)r$rGr&r&r'rOsz$generate_blas_pxd..)r4blas_pxd_preamble)rIbodyr&r&r'generate_blas_pxdsr^aM# 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)NrNcss|]}t|VqdS)N)r[)r$rGr&r&r'rO sz&generate_lapack_pxd..)lapack_pxd_preambler4)rIr&r&r'generate_lapack_pxd sr`z subroutine {name}wrp( + ret, + {argnames} + ) external {wrapper} {ret_type} {wrapper} {ret_type} ret {argdecls} ret = {wrapper}( + {argnames} + ) end z(*)z(ldab,*)z(lda,*)) Zworkabadlr ZduZapeZlldr,)xyrb)ZladivZlanhfZlansfZlapy2Zlapy3cCsNd|kr |St|dd}d|ks,d|kr:||d}n ||d}||S)Nincrerfz(n)r,) special_casesr7)r9funcnamespecialsuffixr&r&r'process_fortran_name*s rmcCs$ddddddg}||kr d|S|S)NZcdotcZcdotuZzdotcZzdotuZcladivZzladivwr&)r9Zincludedr&r&r' called_name5srocsbt}t|\}}d|}fdd|D}dddt||D}tj|||t|dS)Nz , + csg|]}t|qSr&)rm)r$r0)r9r&r'r(Asz+fort_subroutine_wrapper..z css"|]\}}dt||VqdS)z{0} {1}N)r- fortran_types)r$r0r5r&r&r'rOBsz*fort_subroutine_wrapper..)r9wrapperr=argdeclsr1)ror+r4r)fortran_templater-rp)r9r1r*rqtypesrJr=rrr&)r9r'fort_subroutine_wrapper<s   rucCsddd|DS)NrNcss|]}t|VqdS)N)ru)r$rGr&r&r'rOJsz#generate_fortran..)r4)rRr&r&r'generate_fortranIsrvcCs4t|\}}dd|D}dddt||DS)NcSsg|] }t|qSr&)r?)r$r%r&r&r'r(Oszmake_c_args..z, css|]\}}d||VqdS)z{0} *{1}N)r-)r$r5r0r&r&r'rOPszmake_c_args..)r+r4r))r*rtrJr&r&r' make_c_argsMs rwzAvoid F_FUNC({name}wrp, {upname}WRP)({return_type} *ret, {args}); cCs&t|}t|}tj||||dS)N)r9r: return_typer*)rwr?c_func_templater-rB)r9rxr*r&r&r' c_func_declWs rzz&void F_FUNC({name},{upname})({args}); cCst|}tj|||dS)N)r9r:r*)rwc_sub_templater-rB)r9rxr*r&r&r' c_sub_declasr|z#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}|dkrFtj|dt}n tj|d}d|t||tgS)Nr,css|]}t|VqdS)N)rz)r$rGr&r&r'rOsz$generate_c_header..rNcss|]}t|VqdS)N)r|)r$rGr&r&r'rOsLAPACK)lib)r4 c_preambler- lapack_decls cpp_guardc_end)rRrSrIZlib_namerTrUrXr&r&r'generate_c_headers  rcCs.|ddd\}}|d\}}|||fS)N( )r#)rGZ name_and_typer*r1r9r&r&r'split_signaturesrcCsTddttj|D}dd|D}dd|D}tt||tdd}|||fS)NcSsg|]}|r|ds|qS)#) startswith)r$liner&r&r'r(sz filter_lines..cSs&g|]}|dddkrt|qS)rrvoid)r#r)r$rr&r&r'r(scSs&g|]}|dddkrt|qS)rrr)r#r)r$rr&r&r'r(sr)key)mapstrstriplistsortedr)linesrRrSrIr&r&r' filter_liness rcs$ddlmtfdd|DS)Nr)newerc3s.|]&}D]}tj|o"||Vq qdS)N)ospathexists)r$dstsrc)r src_filesr&r'rOszall_newer..)distutils.dep_utilrall)r dst_filesr&)rrr' 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.hc Cstjt||f}|d|d|||d|d||f} ttt|| rVtddSddg} ddd| Dd } dd d| Dd } dd d| Dd } t |d }| }WdQRXt |}t ||f}t |dd }| | | |WdQRXt|d}t |dd }| | | |WdQRXt|d}t |d }| | | |WdQRXt|d}t |d }| | | |WdQRXt |d }| }WdQRXt |}t||f}t |dd }| | | |WdQRXt|d}t |dd }| | | |WdQRXt|d}t |d }| | | |WdQRXt|d}t |d }| | | |WdQRXdS)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. r,cSsg|]}d|dqS)z/* z */ )rstrip)r$rr&r&r'r(szmake_all..rNcSsg|] }d|qS)z# r&)r$rr&r&r'r(scSsg|] }d|qS)zc r&)r$rr&r&r'r(srrnrPr)ZBLAS)r})rrabspath__file__chdirBASE_DIRrprintr4open readlinesrrVwriter^rvrrYr`)Zblas_signature_fileZlapack_signature_fileZ blas_nameZ lapack_nameZblas_fortran_nameZlapack_fortran_nameZblas_header_nameZlapack_header_namerrcommentsZccommentZ pyxcommentZfcommentfZ blas_sigsZblas_pyxZblas_pxdZ blas_fortranZ blas_c_headerZ lapack_sigsZ lapack_pyxZ lapack_pxdZlapack_fortranZlapack_c_headerr&r&r'make_allsv                        r__main__)rrrrrrrr);__doc__ collectionsroperatorrrrrdirnamerrrpr?r+rAr6r.rDrErFrHrKrLrMrQrVrWrYrZr[r\r^r_r`rsZdimsZxy_specialized_dimsZa_specialized_dimsdictrirmrorurvrwryrzr{r|rrrrrrrrr__name__r&r&r&r's    48M          B