:jddlZddlZddlZddlmZmZmZmZddlm Z ddlm Z m Z m Z mZmZmZmZmZmZmZddlZddlmZmZmZddlmZ ddlmZej<ZdZ dd lm"Z#e#jHZejJejLgZ'ejPejRgZ*e'e*zZ+d Z,d Z-d Z.dZ/dZ0dZ1d+dZ2GddZ3GddZ4GddZ5GddZ6GddZ7GddZ8GddZ9Gd d!Z:e2d"d#d$Z;ejxj{d d%&d'Z>ejxj{ed()d*Z?y#e!$rdZd Z YwxYw#e!$rdZ#Y wxYw),N) assert_equalassert_almost_equalassert_array_almost_equalassert_allclose)raises) arangetriutrilzeros tril_indicesonesdiagappendeyenonzero)get_blas_funcstoeplitzsolve) HAS_ILP64)_fblasTF) _fblas_64c 0tdtjdtjdtjdtjdf\}}}t |j dt |j dtd}t |j dtd tj }t |j d td d }t |j d td tj }t |j dtd tjdtj tjdtj f}t |j dy) N)axpyrrrF)dtypeorderCzrotgdgemmrcr) rnpempty complex64 complex128rtypecode clongdoublefloat64)f1f2f3s q/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/linalg/tests/test_blas.pytest_get_blas_funcsr1's  & C 8 & S 9 ; JBBc"c"  Bc" bll 3Bc" c *Bc" bnn 5Bc" & ;& =? Bc"ctdtjdfd}tr|jtj k(sJtj sJtr|jdk(sJ tdtjdf}tr|jtj k(sJtj sJtr|jdk(sJyy)Nr# preferredilp64fblas_64)rr&rr int_dtypeint64int32 module_namer#s r0!test_get_blas_funcs_ilp_preferredr>Ks &266!9,k BD)24>>RXX %@ @@ @-64  z )C CG &266!9, /D)24>>RXX %@ @@ @-64  z )C CG )r2cVtrStdtjdfd}|jtj k(sJ|j dk(sJytjt5tdtjdfddddy#1swYyxYw)Nr#r4Tr6r8) rrr&rr9r:r<pytestr RuntimeErrorr=s r0test_get_blas_funcs_ilp_truerBWs~frvvayl$?~~))):--- ]]< ( = 6BFF1I 6BFF1I > >rCctdtj\}}|jdk(sJ|jdk(sJtdtj\}}}||usJ||usJy)N)nrm2dotr$r%)rKdotcdotu)rr&r(r*r,)fghs r0test_get_blas_funcs_aliasrQmsg / >DAq ::   ::  4BJJGGAq! 6M6 6M6r2c|jdr|dd}n|dvr|d}n |dvr|d}tjtjtjtj d}||S)z%Array dtype from a blas-style prefix.iN)scdzr)cszd)r r%r"s) startswithr&r)r(r,float32)prefixdt_maps r0_dt_from_prefixr^xse  <  < ==r||"**2:: VF &>r2c |tdftdfg}g}|D]\}}|D]}t|}|Btjd||d||tj j |d}n.t|||z} tj| ||d||}|j|tj jd |S) aParametrize a test over BLAS prefixes, "sdcz", and over the BLAS modules, `fblas,fblas_64. Given a func_name "gemm", this generates a pytest parametrization over up to 8 variants: sgemm, dgemm, cgemm, zgemm, each loaded from `fblas` (i.e. 32-bit LP64 variants) and `fblas_64` (i.e. 64-bit ILP64 variants).. If a module is not available, the pytest parameter has a skip mark, so that the test is skipped with a descriptive message. Parameters ---------- func_name : str The base name of a BLAS function, e.g. "gemm" prefixes : str or sequence BLAS prefixes to prepend to `func_name`. E.g. ``func_name='gemm', prefixes='cz'`` generates `cgemm` and `zgemm`. modules : sequence of ``(module, str)`` pairs BLAS modules to fetch functions from. By default, use `fblas` (LP64 variant) and fblas_64 (ILP64 variant). NrEr8. is not availablereason)idmarks)rdzf, dtype) rEr8r^r@parammarkskipgetattrr parametrize) func_nameprefixesmodulesparamsmodmod_namer\rparam_rNs r0parametrize_blasrrs,7#h %;< F " X "F#F+E{%"1VHYK8 ++**XJ>O2P*QC)!34ahZq 3TU MM& ! """ ;; " ":v 66r2cHeZdZedddZedddZedgddZed d d Zed d dZedd dZ edgddZ edgddZ edddZ edgddZ y)TestFBLAS1Simplersdczct|gdgddgd|tvrt|gdgddgdyy)NrTrr4)rr4a) )rT@r4)r|y$@r~rCOMPLEX_DTYPESselfrNrs r0 test_axpyzTestFBLAS1Simple.test_axpys;!!IzQ"?", . N " %a J!&D&4 6 #r2copyct|gddgdzgd|tvrt|gddgdzgdyy)N)r4ryr4)r4@y@@rrs r0 test_copyzTestFBLAS1Simple.test_copys=!!Is1u"5yA N " %a s1u&=} M #r2asum)rYr"rUrVcft|gdd|tvrt|gddyy)Nr4ry @r@)rrrs r0 test_asumzTestFBLAS1Simple.test_asums,AjM2. N " . 12 6 #r2rKsdc2t|gdgddy)Nr)rryrTirrs r0test_dotzTestFBLAS1Simple.test_dotsAj)4b9r2rMczc2t|gdgddy)Nrrr4rTy"@rrs r0 test_dotuzTestFBLAS1Simple.test_dotusAni8%@r2rLc2t|gdgddy)NrrrrTy@,rrs r0 test_dotczTestFBLAS1Simple.test_dotcsAnj95Ar2rJct|gdtjd|tvr(t|gdtjdyy)Nr2r)rmathsqrtrrs r0 test_nrm2zTestFBLAS1Simple.test_nrm2s<AjM499R=9 N " . 1499R= A #r2scal)rYr"rWrXcrt|dgdgd|tvrt|dgdgdyy)Nrr)i r4r)y"@iy"@(rrs r0 test_scalzTestFBLAS1Simple.test_scals2!!Az"2K@ N " %a>&: @!!A1v1v&'VaV$4#69:AB7H J N " %aB7QF&;(*Bx!Q&8 : %a<"b&B()1vBx"b&B D #r2gerurct|dddgddgddgdd ggt|d gd ddgdd gd d gddggy)NrTrrr4rrrrrrrrrr~rrrs r0 test_geruzTestFBLAS2Simple.test_gerus[!!AAwA"7$&8aV"4 6!!B r2h"?$%q6B8b"X"> @r2gercct|dddgddgddgdd ggt|dgd ddgdd gd d gd dggy)NrTrrr4rrrrrrrrr~rrrs r0 test_gerczTestFBLAS2Simple.test_gercs[!!AAwA"7$&8aV"4 6!!A|b"X">$%q6B8b"X"> @r2syrc Jtjddd}tj|ddtjf|z}tj|dddtjf|dddz}tjdddd }tjdd dj d }tj|ddtjf|z}tj|dddtjf|dddz} tj tjd |tjd fj} tj|j} t|d||| t|d|d|j| t|d|ddd || t|d|ddd || |tvrt|d||| t|d|d|j| t|d| ddd || t|d| ddd | | tjd|d} |d|| d} t| || |d|| } | | usJt| d|z| nStjd|d} |d|| d} t| || |d|| } | | usJt| d|z| tt|d|dtt|d|dtt|d|dtt|d|dtt|d|dtt|d|dtt|d|tjdddy) NrTryr"r$rxr!@Fendpointr}Dr?rtolTlowerrincxoffxnrr4rrrr{ overwrite_a@rzrrrr)r&rr newaxislinspaceviewc_r ravelfinfoepsrTr assert_raises Exception)rrNrrresx resx_reverserr resz resz_reversewrr{bs r0test_syrzTestFBLAS2Simple.test_syrs IIa# &wwqBJJ'!+,wwq2rzz!12QttW<= KK3U 3 IIa# & + +C 0wwqBJJ'!+,wwq2rzz!12QttW<= EE"((1+q"((1+- . 4 4 6xx""#q 4d3#q-tvvDA#qqqA64H#qrQ7$4 1 N " Ac1It$ 7 Ac1D1466 E Ac111:Dt L Ac12A;(t 5,A#qA40A At$ /#qAAA: : AqvD 1,A#qA40A At$ /#qAAA: : AqvD 1iC3iC3iC4iCb1iCa0iC!4iCbhhvsC.HIr2herc "tjddd}tjdddjd}tj|ddtjf|j z}tj|dddtjf|dddj z}tj tjd|tjdfj}tj|j}t|d ||| t|d |d |jj | t|d |d dd|| t|d |ddd|| tjd|d} |d || d } t| || |d|| } | | usJt| d |z| tt|d |dtt|d |dtt|d |dtt|d |dtt|d |dtt|d |d tt|d |tjdddy)NrTryr"r$r}rrxrrrTrr4rrrrrrrzrrrrrrr)r&rrr rconjrr rrrrrrr) rrNrrr rehz rehz_reverserrr{rs r0test_herzTestFBLAS2Simple.test_herOs IIa# & IIa# & + +C 0wwqBJJ'!&&(23wwq2rzz!12QttW\\^CD EE"((1+q"((1+- . 4 4 6xx""#q 4d3#q-tvv{{}4H#qqqA64H#qrQ7$4 1 HHVUC ( c1t ,4d+ c1Nzz1T6-iC3iC3iC4iCb1iCa0iC!4iCbhhvsC.HIr2syr2c tjddd}tjddd}tj|ddtjf|z|ddtjf|zz}tj|dddtjf|dddz|dddtjf|dddzz}tjddd d }tj |j }t|d |||| t|d ||d|ddddf| t|d ||d|j| t|d ||dddd|| t|d ||ddddd|ddddf| t|d ||dddd|| tjd|d} |d ||| d} t| || |d||| } | | usJt| d|z| tt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||tjd ddy)!NrTryr"r$r}rxrrrFrrrr4rTrrrrrincyoffyrrrrrrrrrrrzrrrrr) r&rr rrrrrrr rr) rrNrrrresxy resxy_reverseqrr{rs r0 test_syr2zTestFBLAS2Simple.test_syr2ps IIa# & IIa# &!RZZ-(1,qBJJ/?!/CCD$B$ "2 3a"g ="#DbD"**$4"5$B$"?!@A  KK3U 3xx""#q! e$7#q!q)5!RaR=tD#q!40%''E#q!!!!"ED *#q!!!!"Jbqb"1"f D 2 #q!"12BG%D 2 HHVUC ( c1a1$ /5t, c1a1 zz1U7.iCAA6iCAA6iCAB7iCAA6iCAA6iCAB7iCA4iCA3iCAQ7iCA&#s1KLr2her2c tjdddjd}tjdddjd}|ddtjf|j z|ddtjf|j zz}tj |}|dddtjf|dddj z}||dddtjf|dddj zz }tj |}tj tjd|tjdfj}tj tjd|tjdfj}tj|j} t|d |||| t|d ||d |dd dd f| t|d ||d |jj | t|d ||d dd d|| t|d ||d dd dd |dd dd f| t|d ||dddd|| tjd|d} |d ||| d } t| || |d||| } | | usJt| d |z| tt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||dtt|d ||d tt|d ||d tt|d ||dtt|d ||tjdddy) NrTr}r"r$rrrxrrrr4rTrrrrrrrrrzrrryrrrrrr)r&rrrrr rr rrrrrrr) rrNrrrrr uvrr{rs r0 test_her2zTestFBLAS2Simple.test_her2sP IIa# & + +C 0 IIa3 ' , ,S 1!RZZ- 1668+a2:: .>.II$B$ *+a"glln< 4R4+,q2w||~==  . EE"((1+q"((1+- . 4 4 6 EE"((1+q"((1+- . 4 4 6xx""#q! e$7#q!q)5!RaR=tD#q!40%'',,.! # #q!!!!!DD *#q!!!!!qIbqb"1"f D 2 #q!"12AF%D 2 HHVUC ( c1a1$ /5t, c1a1 zz1U7.iCAA6iCAA6iCAB7iCAA6iCAA6iCAB7iCA4iCA3iCAQ7iCAc2 4r2gbmvc Xtjjd}d}d}d}d}tt |j|dzt ||z dz t |j|dzt ||z dz }|j |}t ||zdz|f|} |d| dddf<|d| ddd f<|d | d ddf<|d | d ddf<|j|j |} |j|j |} |d |d} } |}|||||| | | | |  }| |j| z| | zz}t|||||||| | | | | d }| |jj| z| | zz}t||y)Nr|ryrTrr$)rr)rrTr)rrr)rTrr4rr) mrkuklalphar{rrbeta) rrrrrr{rrrtrans) r&random default_rngrrr astyperKrr)rrNrrngrrrrAAbrrrrfuncry2s r0 test_gbmvzTestFBLAS2Simple.test_gbmvsii##D)     VCJJr!t,eAbDFm<CJJr!t,eAbDFm< > HHUO BrE!GQ QUU1X q (!"b)r2z fname,dtypespmvhpmvc |rtnt}||rdnd}tj|dtj j d}d}|j ||fj|}|tvr||j ||fdzz }||dk(r |jn|jjz }t|\} } || | f} |j |j|} |j |j|} td|zj|}td|zj|}|d |d}}t|f|d \}|||| | | | }||j| z|| zz}t!||||d z ||||| dd|| }||ddddfzj|ddd||dddzz}t!|ddd|t#|d|dy)Nr8rErarr4rr-rr'r5r%)rraprrrrT) rrrrrr0rrrrrxr)r8rEr@rhr&rrrrrrr rr rrKrr)rfnamerr7rorprrrr%rAprrxlongylongrrr rr!s r0test_spmv_hpmvzTestFBLAS2Simple.test_spmv_hpmvs   hU ;%*zH KK8*$56 7ii##D)  JJ1v  % %e , N " QF#B& &A EVOQSS3A1 q!tW JJqM  ' JJqM  'qs ""5)QqS   'Dk58txuKH AUrQ!$ ? QUU1X q (!"b)AaCu45Ebqqq2aSbSk! & &uQTT{ 3dU14a4[6H H!"QTT(B/BqE58,r2sprhprc|rtnt}||rdnd}tj|dtj j d}d}|j ||fj|}|tvr||j ||fdzz }||dk(r |jn|jjz }t|\} } || | f} |j |j|} t j|jjd} |d k(rDtd |d \}| | dddfj!| dddfjz|z}n5td |d \}| | dddfj!| dddfz|z}||| | | }t#d|}||| | f<|d k(r|jn||| | f<t%||y)Nr8rErarr4rr7g@r8)r8r5r%)r7)rrr0rr4r4r$)r8rEr@rhr&rrrrrrr rrtyperrKr r)rr1rr7rorprrrr%r2r3rrr r!ry1fs r0 test_spr_hprzTestFBLAS2Simple.test_spr_hpr7s  hU ;%*zH KK8*$56 7ii##D)  JJ1v  % %e , N " QF#B& &A EUNQSS 2A1 q!tW JJqM  '%%**3/ E>"85 LED1d7$' (9::Q>B"85 LED1d7$' 33a7B AUrQ /F%(AqD !&%BGGIRAqD !#r*r2c|rtnt}||rdnd}tj|dtj j d}d}|j ||fj|}|tvrG||j ||fdzz }||jjz }td|d \}n||jz }td |d \}t|\} } || | f} |j |j|} |j |j|} |d }|j| dddfj| dddfjz}||z|jjz}|||| | | }td |}||| | f<|gdj|gdgdf<t||y)Nr8rErarr4r)hpr2r5r%)spr2r)rrrrr0r:r$)rTr4r)rTrr)rrrT)r8rEr@rhr&rrrrrrrr rKr r)rrr7rorprrrr r%r2r3rrrrr!rr<s r0test_spr2_hpr2zTestFBLAS2Simple.test_spr2_hpr2Zs hU ;%*zH KK8*$56 7ii##D)  JJ1v  % %e , N " QF#B& &A OA"9EMED HA"9EMEDA1 q!tW JJqM  ' JJqM  'a JJL1QW:>>!D!G*//*;< < UQVVXZZ  AUa1 4F%(AqD $&yM$6$6$8Iy !!#r*r2tbmvc tjjd}d}d}|j|j|}t ||f|}t |dzD]1}|j||z |t ||z t ||f<3|tvrI|t|xxd|j|dz|z||dzzdzz j|zz cc<t |dz|f|} t |dzD]} t|| | | dz | df<|} | || | } |j|} t| | | || |d } |d|t |t |f<|j|} t| | | || |dd } |jj|} t| | | || |dd } |jjj|} t| | y) Nrrr4r$rTrrr(r(r{rr(r{rrr(r{rrr)r&rrrr r)rrrrrKrrrrrNrrrr(rrsuprrowr rr!s r0 test_tbmvzTestFBLAS2Simple.test_tbmv{sii##D)   JJqM  ' 1a& &1: AC/2zz!C%/@AfQsUmVC^+ , A N " gajMR#**ac1Wa1gqj-A"B"I"I%"PP PMAaC85 )1: .C#A~BtAvst|  . Aq ! UU1X!"b) Aqq )"'(&)VAY  UU1X!"b) Aqq 2 SSWWQZ!"b) Aqq 2 VVXZZ^^A !"b)r2tbsvc tjjd}d}d}|j|j|}t ||f|}t |dzD]1}|j||z |t ||z t ||f<3|tvrI|t|xxd|j|dz|z||dzzdzz j|zz cc<t |dz|f|} t |dzD]} t|| | | dz | df<|} | || | } t||} t| | | || |d } |d|t |t |f<t||} t| | | || |dd } t|j|} t| | | || |dd } t|jj|} t| | y) Ni90rr4r$rTrrrDrErFrG)r&rrrr r)rrrrrrrrrHs r0 test_tbsvzTestFBLAS2Simple.test_tbsvsii##E*   JJqM  ' 1a& &1: AC/2zz!C%/@AfQsUmVC^+ , A N " gajMR#**ac1Wa1gqj-A"B"I"I%"PP PMAaC85 )1: .C#A~BtAvst|  . Aq ! 1a[!"b) Aqq )"'(&)VAY  1a[!"b) Aqq 2 133]!"b) Aqq 2 1668::q !!"b)r2tpmvc0tjjd}d}|j|j|}|tvr4t |j||f|j||fdzz}nt |j||f}t |\}}|||f} |} | || |} |j|} t| | | || |d} |d|t|t|f<|j|} t| | | || |dd} |jj|} t| | | || |dd} |jjj|} t| | y Nrrr)rr0rrT)rr0rr)rr0rrrr) r&rrrrr r rKrrrr rrNrrrrrr%r2r3r rr!s r0 test_tpmvzTestFBLAS2Simple.test_tpmvsiii##D)  JJqM  ' N "SZZA'#**aV*##Q '!"b)r2N)rrrrrrrrrrrr rr"r@rgrjDTYPESr, REAL_DTYPESrr6r=rArKrNrSrVr]ra).0rs00r0rrs8ff%1&1eT" D# Dfd#@$@ fd#@$@ eV$0J%0JdeT"J#J@fd#'M$'MRfd#.4$.4`ff%*&*B [[WtUm4 [[Wf-!*.5!*F [[WtUm4 [[]-'2^'C De65/ D-'5 6e65/ 6-- 5 -@ [[WtUm4 [[]-&1N&B CU5%. C-&4 5U5%. 5-+ 5 +< [[WtUm4 [[Wf-+.5+>ff%#*&#*Jff%#*&#*Jff%*&*@ff%*&*@ff%*&*0ff%*&*_ E 6H D 5s I  I9 I Irc*eZdZedddZy)TestFBLAS3Simpler#ruc t|ddgdgdggt|ddgdgddgdg|tvr3t|ddgdgdggt|ddgdgdd gd gyyrrrs r0 test_gemmzTestFBLAS3Simple.test_gemmAs!!AsRD/SE7;!!AsRD!aS"9C5A N " %aTFRD&9WI; G %aTFRD!bT&BWI N #r2N)rrrrrrhrr2r0rfrf@sff%O&Or2rfcleZdZdZedddZedddZedddZy) TestBLAS3Symmctjddgddgg|_tjgdgdg|_tjd|_tjgdgdg|_y) Nrr)rrl@)rlr)rr4)rrng @)rmrlg"@)r&arrayr{rr r%trs r0 setup_methodzTestBLAS3Symm.setup_methodLsgB88%&<(*+=')*r2symmruc||j|j|jdd}t||j||jj |jd|jdd}t||j||j|jj d|jj dd}t||jj y)Nrr{rr%rrrTr{rrr%rr)r{rsider%rr)r{rr%rrprrrNrress r0 test_symmzTestBLAS3Symm.test_symmUs$&&DFFdffBR@!#tvv.$&&((dffArK!#tvv.$&&DFFHH1r#!#tvvxx0r2ctjt5||j|jdddddy#1swYyxYw)z|`side=1` means C <- B*A, hence shapes of A and B are to be compatible. Otherwise, f2py exception is raised. rT)r{rrrwN)r@rrr{rrs r0test_symm_wrong_sidez"TestBLAS3Symm.test_symm_wrong_sideas; ]]9 % 3 $&& 2 3 3 3s !AA c:||j|j|jdd}tj||j sJ||j|jd|jdd}tj||j rJy)zSYMM only considers the upper/lower part of A. Hence setting wrong value for `lower` (default is lower=0, meaning upper triangle) gives a wrong result. rrurTrvN)r{rr%r&allcloserprxs r0test_symm_wrong_uploz"TestBLAS3Symm.test_symm_wrong_uplojsu $&&DFFdffBR@{{3'''$&&DFF!tvvRbI;;sDFF++++r2N)rrrrrrrrzr|rrr2r0rjrjJsX*ff% 1& 1ff%3&3ff%,&,r2rjcNeZdZdZedddZedddZy) TestBLAS3Syrkctjddgddgddgg|_tjgdgdgdg|_tjd d gd d gg|_y) Nrrlrrm)rrlr)rl@)rr*@@@r)r&ror{rpttrqs r0rrzTestBLAS3Syrk.setup_methodwsnB898%&<()+,((RHI'(r2syrkruc||jd}ttj|tj|j||jdd}ttj |tj |jtj |jj}||jdd|}ttj|tj|j|z||jdd}ttj|tj|jy)Nrr{rrT)r{rr)r{rrr%)r{rr) r{rr&r rpr r shaperrrNrr%c0s r0 test_syrkzTestBLAS3Syrk.test_syrks b !!"''!*bggdffo> b *!"''!*bggdffo> WWTVV\\ " brR 0!"''!*bggdffRi.@A b *!"''!*bggdgg.>?r2ctjt5||jdt j ddddy#1swYyxYw)Nr)ryr)r{rr%)r@rrr{r&r rs r0test_syrk_wrong_czTestBLAS3Syrk.test_syrk_wrong_cs=]]9 % 5 bBGGFO 4 5 5 5s )A  AN)rrrrrrrrrrr2r0rrvs?(ff% @& @ ff%5&5r2rcNeZdZdZedddZedddZy)TestBLAS3Syr2kctjddgddgddgg|_tjddgddgddgg|_tjgdgdgd g|_tjddgdd gg|_y) Nrrlrrrmr)rlrnrm)rnrlrl)rmrlrr)r&ror{rrprrqs r0rrzTestBLAS3Syr2k.setup_methodsB898%&B88r7$%=(')*((RHG%&r2syr2kruc<||j|jd}ttj|tj|j ||j|jdd}ttj |tj |j tj|j j}||j|jdd|}ttj|tj|j |z||j|jdd}ttj|tj|jy)Nrr{rrrT)r{rrr)r{rrrr%)r{rrr) r{rrr&r rpr r rrrs r0 test_syr2kzTestBLAS3Syr2k.test_syr2ks $&& +!"''!*bggdffo> $&&! 4!"''!*bggdffo> WWTVV\\ " $&&r :!"''!*bggdffRi.@A $&&! 4!"''!*bggdgg.>?r2c tjt5||j|jdt j ddddy#1swYyxYw)Nr)r`r)r{rrr%)r@rrr{rr&r rs r0test_syr2k_wrong_cz!TestBLAS3Syr2k.test_syr2k_wrong_csF ]]9 % A $&&bhhw.? @ A A As 4AA N)rrrrrrrrrrr2r0rrsA &gv& @' @gv&A'Ar2rceZdZdZdZedddZedddZeddd Zed dd Z ed dd Z edddZ y)TestSyHez2Quick and simple tests for (zc)-symm, syrk, syr2k.cDtjddgddgg|_y)Nrlyr)r&rosigma_yrqs r0rrzTestSyHe.setup_methods$xx"d"%r!,- r2rszcc||j|jd}ttj|tjddgy)NrrrTrxrrr&r rrxs r0rzzTestSyHe.test_symms9$,,$,,b9!"''#,B0@Ar2hemmc||j|jd}ttj|tjddgy)NrrrTrrxs r0 test_hemmzTestSyHe.test_hemms8$,,$,,b9!"''#,A@r2rc||jd}ttj|tjddgy)Nrrrxrrxs r0rzTestSyHe.test_syrks1$,,b)!"''#,R0ABr2herkc||jd}ttj|tjddgy)NrrrTrrxs r0 test_herkzTestSyHe.test_herks0$,,b)!"''#,A@r2rc||j|jd}ttj|dtjddgzy)Nrrrrxrrxs r0 test_syr2k_zrzTestSyHe.test_syr2k_zrs<$,,$,,b9!"''#,277B83D0DEr2her2kc||j|jd}ttj|dtjddgzy)NrrrrTrrxs r0 test_her2k_zrzTestSyHe.test_her2k_zrs;$,,$,,b9!"''#,277Aq6?0BCr2N) rrr__doc__rrrrrzrrrrrrr2r0rrs<-fd#B$B fd#A$A fd#C$Cfd#A$Agt$F%Fgt$D%Dr2rceZdZdZdZedddZedddZedddZeddd Z y ) TestTRMMz!Quick and simple tests for *trmm.c*tjddgddgg|_tjgdgdg|_tjgdgdgdgd gd |_tjd d gddgddgddgddggd |_y)Nrrrrmrrnrrr)rTrTrr4)rrTrry)rrrTr)rrrrTrN)rrTrrryr4rr|rr}r)r&ror{ra2b2rqs r0rrzTestTRMM.setup_methodsB:9&'=(*+((L(((*256((QFQFQFQFQGD!$&r2trmmsdzcc |}tt|d|j|j|d|jj ||jj |d}|jj d}t ||j|jd|d|fzddtj|jzy)NrrT)rwrld)ratol) rrrrrrrr&rr)rrNrrryr(s r0 test_sidezTestTRMM.test_sidesisDGGTWW=3u-tww~~e/D GGMM! TWWtwwrr2A2v6R %!4!44 6r2ruc|d|j|j}tjgdgdg}t ||y)Nr)rg0@grr{rr&rorrrNrresultexpecteds r0test_abzTestTRMM.test_abs:2tvvtvv&88_,./!&(3r2c|d|j|jd}tjgdgdg}t ||y)NrTrr)rnrrlrrs r0 test_ab_lowerzTestTRMM.test_ab_lower s<2tvvtvvT288_,./!&(3r2c|jj|}dD]d}|j}|d|j||}|jj rJt j||rJt||ft j|j}|d|j|d}|jj sJt j||sJt||y)N)TFr) overwrite_bT) rrrr{flags f_contiguousr&may_share_memoryrasfortranarrayr)rrNrroverwrbcopyrs r0test_b_overwriteszTestTRMM.test_b_overwritess FFMM% # #FFFHEr4665f=F{{// //**5&9 99  "  #!!!&&(+2tvvu$7{{''''""5&111!%0r2N) rrrrrrrrrrrrrr2r0rrsv+ &ff%6&6ff%4&4ff%4&4 ff%1&1r2rtrsmruc4tjjd}tj|jdz}|}|jdj |}|jdj |}|d}t t||||t t|||j|d}d} |d}|tvr7|j| | f|j| | fd zzt| z}n|j| | ft| z}|j |}t|} t|} |j| |fj |} |j|| fj |} |||| }t| j|jt| || z}t!||| |||| d }t| j|| z}t!||| |||| d }t| j#j|| z}t!||| |||| d}|d| t%| t%| f<t| || z}t!||| |||| dd}t| j#j|| j#jz}t!||j#j| |||| ddd}|d| t%| t%| f<t| j#j|| j#jz}t!||j#j| y)Nri)rryrrTrr|gr)rr{r)r)rr{rtrans_ar)rr{rr)rr{rrrw)rr{rrrwr)r&rrrrrrrrrrr r rrrrrr)rNrrtolr rBrrrAuAlB1B2rx2s r0 test_trsmr%s ))   %C ((5/  d "C D 6!!%(A 6!!%(A !HE)T5!Q/)T5!##q1 A A $KE  ZZA #**aV"4R"7 73q6 A JJ1v Q ' A aB aB QF  " "5 )B QF  " "5 )B EQ" %B288$ r58 BB% EQ"a 0B rttU2X BB% EQ"a 0B rwwy{{E"H %BB% EQ"1 -B$QxBvay&) r58 BB% EQ"11 5B rwwy{{E"'')++- .BB #. EQ"11A >B$QxBvay&) rwwy{{E"'')++- .BB #.r2zgh-16930)runrcctjdd}tjjj |ddd}t jd}t||y)Nrr}ryr4rxi) r&repeatscipylinalgblasdnrm2rrr)ractualrs r0test_gh_169309r[sK "aA \\   $ $Q1b 1Fyy~HFH%r2zdoes not fail with ILP64 MKLrbctjdd}d}tjjj d|fd}t t5||dd|dddy#1swYyxYw) Nrr}rxrJr5r6ryr4)r&rrrrrr FBLAS_ERROR)rrrs r0test_dnrm2_neg_incxrdsg "aA D LL   , ,VaT , ME { # aAts  A**A3)N)@rr@numpyr& numpy.testingrrrrrrrr r r r r rrrrr scipy.linalgrrrscipy.linalg.blasrrrE __fblas_errorrrF ImportErrorrr8__fblas_64_errorr[r,rcr(r)rrbr1r>rBrHrQr^rrrtrrfrjrrrrrrgxfailrrrr2r0rs  GG*/// 88',%%KN 2++K zz2::& ,, . ~ %!#H D=> +7\?.?.HF *F *ROO),),X 5 5H!A!AJ%D%DP;1;1|&&!2/"2/ju$&&&&9%CDE]" ENHs$ E$!E4$ E10E14E?>E?