o иi eЇ;у@sddlZddlZddlmZmZmZmZmZddl m Z mZmZzddl Z dZWney3dZYnwzddlZeабZdej_dZWneyOdZYnwzddlZdejd<ddlZdZWneyldZYnwzddlZdZWneyАdZYnwzddlZdZWneyФdZYnwzddlZdZ WneyиdZ YnwgdвZ!ej"j#ed d Нej"а$de!бdd ДГГZ%ej"j#ed d Нej"а$dddhddhddhgбddДГГZ&ej"j#ed d Нej"а$de!бddДГГZ'ej"j#edd Нej"а$de!бddДГГZ(ej"j#edd Нej"а$dddhddhddhgбddДГГZ)ej"j#edd Нej"а$de!бddДГГZ*ej"j#edd Нej"а$de!бddДГГZ+ej"j#edd Нej"а$dddhddhddhgбd d!ДГГZ,ej"j#ed"d Нej"а$de!бd#d$ДГГZ-ej"j#ed"d Нej"а$dddhddhddhgбd%d&ДГГZ.ej"j#ed"d Нd'd(ДГZ/ej"j#e d)d Нd*d+ДГZ0ej"а$de!бd,d-ДГZ1ej"а$de!бd.d/ДГZ2ej"j#ed0d Нej"а$de!бd1d2ДГГZ3ej"j#ed0d Нej"а$dddhddhddhgбd3d4ДГГZ4d5d6ДZ5ej"а$de!бd7d8ДГZ6dS)9щN)┌backends┌contract┌contract_expression┌helpers┌sharingй┌Shaped┌ infer_backend┌ parse_backendTF┌GNU┌MKL_THREADING_LAYER)z ab,bc->cazabc,bcd,deazabc,def->fedcbazabc,bcd,df->fazijk,ikjzi,j->ijz ijk,k->ijz AB,BC->CAzTensorflow not installed.)┌reason┌stringcCs╞tа|б}t|g|вRdddЬО}tа|б}ddД|DГ}t|g|вRddiО}tjtdН}|а бП||d|d ЬОWdГn1sEwY|а бtа||бsVJВd dД|DГ}||ОdS)NFй┌optimize┌use_blascSєg|]}|jСqSйй┌shapeй┌.0┌vrr·M/var/www/myenv/lib/python3.10/site-packages/opt_einsum/tests/test_backends.py┌ Dєz#test_tensorflow..rTй┌config┌ tensorflow)┌backend┌outcSєg|]}tа|бСqSr)r┌ to_tensorflowйr┌viewrrrrOє)r┌build_viewsr┌np┌ empty_liker┌tf┌Session┌ _TF_CONFIG┌ as_default┌close┌allclose)r┌views┌ein┌opt┌shps┌expr┌sess┌tensorflow_viewsrrr┌test_tensorflow=s r6┌ constantsщщc sd}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}tjt d На бП|ИddН}WdГn1sUwYtd dД|jdDГГshJВtа ||бspJВ|ИddН}tа ||бs~JВ|tаИбГ}t|tjГsНJВdS)N·ij,jk,kl->liййr9щйr=щ)r?щ╛rr8r9cє(g|]\}}|Иvrtjj|Оn|СqSrйr'┌random┌randйr┌i┌shpйr7rrrYє(z2test_tensorflow_with_constants..c3є$Б|] }|ИvrИ|nИVqdSйNrйrrGйr7┌ops┌varrr┌ [єА"z1test_tensorflow_with_constants..r=r7rrйrcsє$Б|] }|dup t|ГdkVqdS)Nrйr йr┌arrayrrrrQbsА ┌numpy)┌ enumerater'rDrEr┌rangerr)r*r+r,┌all┌_evaluated_constantsr.rr"┌ isinstance┌Tensorй r7┌eq┌shapes┌ non_const┌res_expr3┌res_got┌res_got2┌res_got3rrNr┌test_tensorflow_with_constantsSs$& rgc Cs0tа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}tjtdН}|абПDtа бП0}||dd iО}tа б|usBJВt|Г}|d ksLJВ||dd iО} t|Г|ks[JВWdГn1sewYWdГn1stwYtddД|а бDГГsЖJВtа||бsОJВtа|| бsЦJВdS) NFrcSrrrrrrrrurz0test_tensorflow_with_sharing..rTrrrrcssБ|] }t|tjГVqdSrL)r]r)r^йr┌trrrrQВsАz/test_tensorflow_with_sharing..)rr&rrr)r*r+r,r┌shared_intermediates┌get_sharing_cache┌lenr[┌valuesr'r.) rr/r0r2r3r4┌cache┌tfl1┌cache_sz┌tfl2rrr┌test_tensorflow_with_sharingos$ ·АrrzTheano not installed.cCsРtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бs2JВd dД|DГ}||О}t|tjj ГsFJВdS) NFrcSrrrrrrrrНrztest_theano..rTr┌theanocSr!r)r┌ to_theanor#rrrrХr%) rr&rrr'r.r]rs┌tensor┌TensorVariable)rr/r0r2r3r1┌theano_views┌ theano_optrrr┌test_theanoИs ryztheano not installed.c sЁd}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsPJВtа ||бsXJВ|Иd dН}tа ||бsfJВ|t аИбГ}t|t jjГsvJВdS)Nr:r;rAcrBrrCrFrIrrrаrJz.test_theano_with_constants..c3rKrLrrMrNrrrQвrRz-test_theano_with_constants..r=r7rsrScsrT)NrsrUrVrrrrQиrRrX)rYr'rDrErrZrr[r\r.rrtr]rsrurvr_rrNr┌test_theano_with_constantsЪs&rzc Cs№tа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}tабП0}||ddiО}tаб|us7JВt|Г}|d ksAJВ||ddiО}t|Г|ksPJВWdГn1sZwYtd dД|а бDГГslJВt а||бstJВt а||бs|JВdS)NFrcSrrrrrrrr║rz,test_theano_with_sharing..rTrrsrcssБ|] }t|tjjГVqdSrL)r]rsrurvrhrrrrQ┼sАz+test_theano_with_sharing..)rr&rrrrjrkrlr[rmr'r.) rr/r0r2r3rn┌thn1rp┌thn2rrr┌test_theano_with_sharing┤s ·r}zCupy not installed.cCsдtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бs2JВd dД|DГ}||О}t|tjГsEJВtа|tа |ббsPJВdS) NFrcSrrrrrrrr╨rztest_cupy..rTr┌cupycSr!r)r┌to_cupyr#rrrr╪r%) rr&rrr'r.r]r~┌ndarray┌asnumpy)rr/r0r2r3r1┌ cupy_views┌cupy_optrrr┌ test_cupy╦s rДc sd}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsPJВtа ||бsXJВ|Иd dН}tа ||бsfJВ|t аИбГ}t|t j ГsuJВtа ||аббsJВdS)Nr:r;rAcrBrrCrFrIrrrфrJz,test_cupy_with_constants..c3rKrLrrMrNrrrQцrRz+test_cupy_with_constants..r=r7r~rScsrT)Nr~rUrVrrrrQэrRrX)rYr'rDrErrZrr[r\r.r~┌asarrayr]rА┌getr_rrNr┌test_cupy_with_constants▐s&rЗzjax not installed.cCsxtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бs2JВt|tjГs:JВdS) NFrcSrrrrrrrr rztest_jax..rTr┌jax)rr&rrr'r.r]rА)rr/r0r2r3r1rrr┌test_jax·s rЙcs┤d}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsPJВtа ||бsXJВdS) Nr:r;rAcrBrrCrFrIrrrrJz+test_jax_with_constants..c3rKrLrrMrNrrrQrRz*test_jax_with_constants..r=r7rИrScsrT)NrИrUrVrrrrQrR) rYr'rDrErrZrr[r\r.)r7r`rarbrcr3rdrrNr┌test_jax_with_constantss&rКc s╞d}d}ddД|DГ}t|g|вRОЙИ|О}tаИб}||Оаб}|tj|ddНks-JВtаtаЗfddДбб}||Г}td d Дt||ГDГГsKJВddДt||ГDГ}||Оаб} | |ksaJВdS)N· ij,jk,kl->йr<r>)r?r9cSєg|]}tjj|ОСqSrйr'rD┌randnйr┌srrrr єz)test_jax_jit_gradient..gёhуИ╡°ф>)┌relcєИ|ОSrLrйr/йr3rr┌)єz'test_jax_jit_gradient..csє Б|]\}}|j|jkVqdSrLrйr┌v1┌v2rrrrQ+єАz(test_jax_jit_gradient..cSєg|] \}}|d|СqSйg№йё╥MbP?rйrr┌dvrrrr.є) rrИ┌jit┌item┌pytest┌approx┌gradr[┌zip) r`rar/┌x0┌jit_expr┌x1┌ grad_expr┌ view_grads┌ new_views┌x2rrЦr┌test_jax_jit_gradients r░zautograd not installed.csРd}d}ddД|DГ}t|g|вRОЙИ|О}tаЗfddДб}||Г}tddДt||ГDГГs2JВd dДt||ГDГ}И|О}||ksFJВdS) NrЛrМcSrНrrОrРrrrr7rТz*test_autograd_gradient..crФrLrrХrЦrrrЧ<rШz(test_autograd_gradient..csrЩrLrrЪrrrrQ>rЭz)test_autograd_gradient..cSrЮrЯrrаrrrrArв)r┌autogradrзr[rи)r`rar/rйrмrнrоrлrrЦr┌test_autograd_gradient3sr▓cs╩tаdбЙtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}ЗfddД|DГ}||О}t|ИjГs=JВtа |tа |ббsHJВt|g|вRО}t|ИjГsXJВtа |tа |ббscJВdS) Nz dask.arrayFrcSrrrrrrrrLrztest_dask..rTcsg|] }Иj|ddНСqS)r9)┌chunks)┌ from_arrayйr┌xй┌darrrPs)rе┌importorskiprr&rrr]┌Arrayr'r.rW)rr/r0r2r3┌da_views┌da_optrr╖r┌ test_daskFs r╜c stаdбЙtа|б}|D]}tjаdбtjаddg|jdddgб}d||<qt |g|вRdddЬО}d d Д|DГ}t |g|вRddiО}Зfdd Д|DГ}||О}t|ИjГsZJВtа ||аббsdJВt |g|вRО}t|ИjГstJВtа ||аббs~JВdS) N┌sparseщ*FTgЪЩЩЩЩЩй?gffffffю?rrcSrrrrrrrrkrztest_sparse..rcsg|]}Иjа|бСqSr)┌COO┌ from_numpyr╡йr╛rrrorТ)rеr╣rr&r'rD┌seed┌choicerrrr]r└r.┌todense) rr/r$┌maskr0r2r3┌sparse_views┌ sparse_optrr┬r┌test_sparse^s r╔zTorch not installed.cCsжtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бs2JВd dД|DГ}||О}t|tjГsEJВtа||а ба ббsQJВdS) NFrcSrrrrrrrrГrztest_torch..rTr┌torchcSr!r)r┌to_torchr#rrrrЛr%)rr&rrr'r.r]r╩r^┌cpurX)rr/r0r2r3r1┌torch_views┌ torch_optrrr┌ test_torch}s r╧c sd}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsPJВtа ||бsXJВ|Иd dН}tа ||бsfJВ|t аИбГ}t|t jГsuJВ|jjdkr|абn|абаб}tа ||бsНJВdS)Nr:r;rAcrBrrCrFrIrrrЧrJz-test_torch_with_constants..c3rKrLrrMrNrrrQЩrRz,test_torch_with_constants..r=r7r╩rScsrT)Nr╩rUrVrrrrQЯrRrXr╠)rYr'rDrErrZrr[r\r.rr╦r]r╩r^┌device┌typerXr╠r_rrNr┌test_torch_with_constantsСs & r╥cCs0tdГ}t|ГdksJВt|gdГdksJВdS)N)r8r9r=┌ opt_einsum┌autorXr)r╢rrr┌+test_auto_backend_custom_array_no_tensordotнsr╒cCs╥tа|б}t|g|вRdddЬО}|jtksJВddД|DГ}t|g|вRddiО}ddД|DГ}t|g|вRdd iО}|jtksCJВtа||аt ббsNJВ||dd iО}|jtks\JВtа||аt ббsgJВdS) NFrcSrrrrrrrr║rz.test_object_arrays_backend..rTcSsg|]}|аtбСqSr)┌astype┌objectr#rrrr╜r%rr╫) rr&r┌dtyper╫rr'r.r╓┌float)rr/r0r2r3┌ obj_views┌obj_optrrr┌test_object_arrays_backend┤s r▄)7rXr'rеr╙rrrrr┌opt_einsum.contractrr r r~┌ found_cupy┌ImportErrorrr)┌ConfigProtor+┌gpu_options┌allow_growth┌found_tensorflow┌os┌environrs┌found_theanor╩┌found_torchrИ┌ found_jaxr▒┌found_autograd┌tests┌mark┌skipif┌parametrizer6rgrrryrzr}rДrЗrЙrКr░r▓r╜r╔r╧r╥r╒r▄rrrr┌s╕