o ░2·h╩<у@s2ddlmZddlZddlmZmZmZmZddlm Z m Z mZddlm Z z-zddlmZddlmZWneyJddlmZddlmZYnweГZdej_Wn ey\Ynwgd вZejаd eбd eddfdd ДГZejаdddhddhddhgбdeeddfddДГZejаd eбd eddfddДГZejаd eбd eddfddДГZejаdddhddhddhgбdeeddfddДГZ ejаd eбd eddfddДГZ!ejаd eбd eddfddДГZ"ejаdddhddhddhgбdeeddfddДГZ#ejаd eбd eddfdd ДГZ$ejаdddhddhddhgбdeeddfd!d"ДГZ%d3d#d$ДZ&d3d%d&ДZ'ejаd eбd eddfd'd(ДГZ(ejаd eбd eddfd)d*ДГZ)ejаd eбd eddfd+d,ДГZ*ejаdddhddhddhgбdeeddfd-d.ДГZ+d3d/d0ДZ,ejаd eбd eddfd1d2ДГZ-dS)4щ)┌SetN)┌backends┌contract┌contract_expression┌sharingй┌ArrayShaped┌ infer_backend┌ parse_backend)┌build_views)┌ConfigProto)┌SessionT)z ab,bc->cazabc,bcd,deazabc,def->fedcbazabc,bcd,df->fazijk,ikjzi,j->ijz ijk,k->ijz AB,BC->CA┌string┌returnc Cs╓tаdб}tаdбt|Г}t|g|вRdddЬО}|а|б}ddД|DГ}t|g|вRddiО}ttd Н}|абП||d|d ЬОWdГn1sMwY|а б|а ||бs^JВddД|DГ}||ОdS)N┌numpy┌ tensorflowFй┌optimize┌use_blascSєg|]}|jСqSйй┌shapeй┌.0┌vrr·Y/var/www/html/chatgem/venv/lib/python3.10/site-packages/opt_einsum/tests/test_backends.py┌ -єz#test_tensorflow..rTй┌config)┌backend┌outcSєg|]}tа|бСqSr)r┌ to_tensorflowйr┌viewrrrr8є)┌pytest┌importorskiprr┌ empty_liker┌ TFSession┌ _TF_CONFIG┌ as_default┌close┌allclose) r┌np┌views┌ein┌opt┌shps┌expr┌sess┌tensorflow_viewsrrr┌test_tensorflow$s r8┌ constantsщщc s2tаdбЙtаdб}d}d}hdгИ\}ЗЗfddДt|ГDГЙИjj||ОЙt|gЗЗЗfdd Дtd ГDГвRО}t|gИвRdИiО}tt dНа бП|Иdd Н}WdГn1s_wYtdd Д|jdDГГsrJВИа ||бszJВ|Иdd Н}Иа ||бsИJВ|tаИбГ} t| |jГsЧJВdS)Nrr·ij,jk,kl->liййr;щйr?щ)rAщ╛rr:r;cє(g|]\}}|ИvrИjj|Оn|СqSrй┌random┌randйr┌i┌shpйr9r0rrrDє(z2test_tensorflow_with_constants..c3є$Б|] }|ИvrИ|nИVqdSйNrйrrIйr9┌ops┌varrr┌ FєА"z1test_tensorflow_with_constants..r?r9rйr!csє$Б|] }|dup t|ГdkVqdS)Nrйr йr┌arrayrrrrSMsА )r(r)┌ enumeraterFrGr┌rangerr+r,r-┌all┌_evaluated_constantsr/rr$┌ isinstance┌Tensor) r9┌tf┌eq┌shapes┌ non_const┌res_expr5┌res_got┌res_got2┌res_got3rйr9r0rQrRr┌test_tensorflow_with_constants<s( & ric sDtаdб}tаdбЙt|Г}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}ttd Н}|абПDtа бП0}||d diО}tа б|usJJВt|Г} | dksTJВ||d diО} t|Г| kscJВWdГn1smwYWdГn1s|wYtЗfdd Д|а бDГГsРJВ|а||бsШJВ|а|| бsаJВdS)NrrFrcSrrrrrrrrcrz0test_tensorflow_with_sharing..rTrr!rc3sБ|] }t|ИjГVqdSrN)r^r_йr┌tйr`rrrSpsАz/test_tensorflow_with_sharing..)r(r)rrrr+r,r-r┌shared_intermediates┌get_sharing_cache┌lenr\┌valuesr/)rr0r1r2r4r5r6┌cache┌tfl1┌cache_sz┌tfl2rrlr┌test_tensorflow_with_sharing[s( ·Аruc Csвtаdб}tаdб}t|Г}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||d diО}|а||бs;JВd dД|DГ}||О} t| |jjГsOJВdS)Nr┌theanoFrcSrrrrrrrr}rztest_theano..rTr!cSr#r)r┌ to_theanor%rrrrЕr') r(r)rrrr/r^┌tensor┌TensorVariable) rr0rvr1r2r4r5r3┌theano_views┌ theano_optrrr┌test_theanovs r|c stаdбЙtаdб}d}d}hdгИ\}ЗЗfddДt|ГDГЙИjj||ОЙt|gЗЗЗfdd Дtd ГDГвRО}t|gИвRdИiО}|ИddН}td d Д|j dDГГs[JВИа ||бscJВ|ИddН}Иа ||бsqJВ|tаИбГ} t | |jjГsБJВdS)Nrrvr<r=rCcrDrrErHrKrrrТrLz.test_theano_with_constants..c3rMrNrrOrPrrrSФrTz-test_theano_with_constants..r?r9rUcsrV)NrvrWrXrrrrSЪrT)r(r)rZrFrGrr[rr\r]r/rrwr^rxry) r9rvrarbrcrdr5rerfrgrrhr┌test_theano_with_constantsКs &r}c stаdб}tаdбЙt|Г}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}tабП0}||d diО}tаб|us@JВt|Г}|d ksJJВ||d diО} t|Г|ksYJВWdГn1scwYt ЗfddД|а бDГГswJВ|а||бsJВ|а|| бsЗJВdS) NrrvFrcSrrrrrrrrоrz,test_theano_with_sharing..rTr!rc3sБ|] }t|ИjjГVqdSrN)r^rxryrjйrvrrrS╣sАz+test_theano_with_sharing..)r(r)rrrrrmrnror\rpr/) rr0r1r2r4r5rq┌thn1rs┌thn2rr~r┌test_theano_with_sharingжs" ·rБc Cs╢tаdб}tаdб}t|Г}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||d diО}|а||бs;JВd dД|DГ}||О} t| |jГsNJВ|а||а| ббsYJВdS)Nr┌cupyFrcSrrrrrrrr╞rztest_cupy..rTr!cSr#r)r┌to_cupyr%rrrr╬r') r(r)rrrr/r^┌ndarray┌asnumpy) rr0rВr1r2r4r5r3┌ cupy_views┌cupy_optrrr┌ test_cupy┐s rИc stаdбЙtаdб}d}d}hdгИ\}ЗЗfddДt|ГDГЙИjj||ОЙt|gЗЗЗfdd Дtd ГDГвRО}t|gИвRdИiО}|ИddН}td d Д|j dDГГs[JВИа ||бscJВ|ИddН}Иа ||бsqJВ||аИбГ} t| |j ГsАJВИа || аббsКJВdS)NrrВr<r=rCcrDrrErHrKrrr▄rLz,test_cupy_with_constants..c3rMrNrrOrPrrrS▐rTz+test_cupy_with_constants..r?r9rUcsrV)NrВrWrXrrrrSхrT)r(r)rZrFrGrr[rr\r]r/┌asarrayr^rД┌get) r9rВrarbrcrdr5rerfrgrrhr┌test_cupy_with_constants╘s" &rЛcCsКtаdб}tаdбt|Г}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||d diО}|а||бs;JВt||jГsCJВdS) Nr┌jaxFrcSrrrrrrrr∙rztest_jax..rTr!)r(r)rrrr/r^rД)rr0r1r2r4r5r3rrr┌test_jaxЄs rНcsрtаdбЙИjаdбЙd}d}hdгИ\}ЗЗЗfddДt|ГDГЙИjаИ||бЙt|gЗЗЗfdd Дtd ГDГвRО}t|gИвRdИiО}|ИddН}t d d Д|j dDГГs^JВИjаИjа ||ббdksnJВdS)NrМщ*r<r=rCcs*g|]\}}|ИvrИjаИ|бn|СqSrйrF┌uniformrH)r9rМ┌keyrrr s*z+test_jax_with_constants..c3rMrNrrOrPrrrSrTz*test_jax_with_constants..r?r9rUcss&Б|]}|dupt|ГаdбVqdS)NrМ)r ┌ startswithrXrrrrSsА$g:М0тОyE>)r(r)rF┌PRNGKeyrZrРrr[rr\r]r┌sum┌abs)r9rarbrcrdr5rer)r9rМrСrQrRr┌test_jax_with_constantss &$rЦc sтtаdбЙИjаdбЙd}d}ЗЗfddД|DГ}t|g|вRОЙИ|О}ИаИб}||Оаб}|tj|ddНks;JВИаИаЗfd d Дбб}||Г}t ddДt ||ГDГГsYJВd dДt ||ГDГ}||Оаб} | |ksoJВdS)NrМrО· ij,jk,kl->йr>r@)rAr;csg|] }ИjаИ|бСqSrrПйr┌s)rМrСrrrєz)test_jax_jit_gradient..gёhуИ╡°ф>)┌relcєИ|ОSrNrйr1йr5rr┌&єz'test_jax_jit_gradient..csє Б|]\}}|j|jkVqdSrNrйr┌v1┌v2rrrrS(єАz(test_jax_jit_gradient..cSєg|] \}}|d|СqSйg№йё╥MbP?rйrr┌dvrrrr+є)r(r)rFrУr┌jit┌item┌approx┌gradr\┌zip) rarbr1┌x0┌jit_expr┌x1┌ grad_expr┌ view_grads┌ new_views┌x2r)r5rМrСr┌test_jax_jit_gradients r╕c sиtаdбЙtаdб}d}d}ЗfddД|DГ}t|g|вRОЙИ|О}|аЗfddДб}||Г}td d Дt||ГDГГs>JВddДt||ГDГ}И|О}||ksRJВdS)Nr┌autogradrЧrШcsg|]}Иjj|ОСqSr)rF┌randnrЩ)r0rrr6єz*test_autograd_gradient..crЭrNrrЮrЯrrrа;rбz(test_autograd_gradient..csrвrNrrгrrrrS=rжz)test_autograd_gradient..cSrзrиrrйrrrr@rл)r(r)rrпr\r░) r╣rarbr1r▒r┤r╡r╢r│r)r5r0r┌test_autograd_gradient0s r╝cs╥tаdб}tаdбЙt|Г}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}Зfd dД|DГ}||О}t|ИjГsAJВ|а||а|ббsLJВt|g|вRО}t|ИjГs\JВ|а||а|ббsgJВdS) Nrz dask.arrayFrcSrrrrrrrrLrztest_dask..rTcsg|] }Иj|ddНСqS)r;)┌chunks)┌ from_arrayйr┌xй┌darrrPrЫ) r(r)rrrr^┌Arrayr/rY)rr0r1r2r4r5┌da_views┌da_optrr┴r┌ test_daskEs r╞c s$tаdб}tаdбЙt|Г}|D]}|jаdб|jаddg|jdddgб}d||<qt|g|вRddd ЬО}d dД|DГ}t|g|вRddiО}Зfd dД|DГ}||О} |jsdtа |бdksbJВdSt | ИjГslJВ|а|| а ббsvJВt|g|вRО} t | ИjГsЖJВ|а|| а ббsРJВdS)Nr┌sparserОFTgЪЩЩЩЩЩй?gffffffю?rrcSrrrrrrrrlrztest_sparse..rcsg|]}Иjа|бСqSr)┌COO┌ from_numpyr┐йr╟rrrpr╗g)r(r)rrF┌seed┌choicerrrrоr^r╚r/┌todense) rr0r1r&┌maskr2r4r5┌sparse_views┌ sparse_optrr╩r┌test_sparse^s( r╤c Csвtаdб}t||jdН}|j|g|вRО}ddД|DГ}t|g|вRddiО}||ddiО}|jа||бddД|DГ}||О}t||j ГsHJВ|jа||бdS) N┌torch)┌array_functioncSrrrrrrrrЙrztest_torch..rTr!cSr#r)r┌to_torchr%rrrrРr') r(r)rrG┌einsumr┌testing┌assert_closer^r_) rr╥r1r2r4r5r3┌torch_views┌ torch_optrrr┌ test_torchВs r┌c stаdбЙd}d}hdгИ\}ЗЗfddДt|ГDГЙИj||ОЙt|gЗЗЗfddДtd ГDГвRd diО}t|gИвRdИiО}|ИddН}td dД|jdDГГsXJВИj а ||б|ИddН}Иj а ||б|tаИбГ}t |ИjГs{JВИj а ||бdS)Nr╥r<r=rCcs&g|]\}}|ИvrИj|Оn|СqSr)rGrH)r9r╥rrrЭs&z-test_torch_with_constants..c3rMrNrrOrPrrrSЯrTz,test_torch_with_constants..r?r!r9rUcsrV)Nr╥rWrXrrrrSеrT)r(r)rZrGrr[rr\r]r╓r╫rr╘r^r_) r9rarbrcrdr5rerfrgr)r9rQr╥rRr┌test_torch_with_constantsЦs ,r█cCsDtdГ}t|ГdksJВt|gdГdksJВt|gdГdks JВdS)N)r:r;r?┌ opt_einsum┌autorr)r└rrr┌+test_auto_backend_custom_array_no_tensordot▓sr▐cCs┌tаdб}t|Г}t|g|вRdddЬО}|jtksJВddД|DГ}t|g|вRddiО}ddД|DГ}t|g|вRd d iО}|jtksGJВ|а||аt ббsRJВ||d d iО}|jtks`JВ|а||аt ббskJВdS)NrFrcSrrrrrrrr┴rz.test_object_arrays_backend..rTcSsg|]}|аtбСqSr)┌astype┌objectr%rrrr─r'r!rр) r(r)rr┌dtyperрrr/r▀┌float)rr0r1r2r4r5┌ obj_views┌obj_optrrr┌test_object_arrays_backend║s rх)rN).┌typingrr(r▄rrrr┌opt_einsum.contractrr r ┌opt_einsum.testingrrrr r+┌ImportError┌tensorflow.compat.v1r,┌gpu_options┌allow_growth┌tests┌mark┌parametrize┌strr8┌intrirur|r}rБrИrЛrНrЦr╕r╝r╞r╤r┌r█r▐rхrrrr┌sh■ #