a o=icЇ;у@sddlZddlZddlmZmZmZmZmZddl m Z mZmZzddl Z dZWneyfdZYn0z ddlZeабZdej_dZWneyЮdZYn0z"ddlZdejd<ddlZdZWney╪dZYn0zddlZdZWneРydZYn0zddlZdZWneРy,dZYn0zddlZdZ WneРyVdZ Yn0gdв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_backendTFZGNUZMKL_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Гn1sК0Y|а бtа||бsмJВd dД|DГ}||ОdS)NFй┌optimizeZuse_blascSsg|] }|jСqSйй┌shapeй┌.0┌vrr·o/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/opt_einsum/tests/test_backends.py┌ Dєz#test_tensorflow..rTй┌config┌ tensorflow)┌backend┌outcSsg|]}tа|бСqSr)r┌ to_tensorflowйr┌viewrrrrOr)r┌build_viewsr┌npZ empty_liker┌tf┌Session┌ _TF_CONFIG┌ as_default┌close┌allclose)r┌views┌ein┌opt┌shps┌expr┌sessZtensorflow_viewsrrr┌test_tensorflow=s .r.┌ constantsщщc s d}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}tjt d На бП|ИddН}WdГn1sк0Ytd dД|jdDГГs╨JВtа ||бsрJВ|ИddН}tа ||бs№JВ|tаИбГ}t|tjГРsJВdS)N·ij,jk,kl->liййr1щйr5щ)r7щ╛rr0r1cs(g|] \}}|Иvr tjj|Оn|СqSrйr!┌random┌randйr┌iZshpйr/rrrYrz2test_tensorflow_with_constants..c3s"|]}|ИvrИ|nИVqdSйNrйrr>йr/┌ops┌varrr┌ [rz1test_tensorflow_with_constants..r5r/rrйrcss"|]}|dupt|ГdkVqdS)Nrйr йr┌arrayrrrrEbs ┌numpy)┌ enumerater!r;r<r┌rangerr"r#r$r%┌all┌_evaluated_constantsr'rr┌ isinstance┌Tensorй r/┌eq┌shapes┌ non_const┌res_expr,┌res_gotZres_got2Zres_got3rrBr┌test_tensorflow_with_constantsSs"&* rWc Cs6tа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}tjtdН}|абПИtа бП`}||dd iО}tа б|usДJВt|Г}|d ksШJВ||dd iО} t|Г|ks╢JВWdГn1s╩0YWdГn1sш0YtddД|а бDГГРsJВtа||бРs JВtа|| бРs2JВdS) NFr cSsg|] }|jСqSrrrrrrrurz0test_tensorflow_with_sharing..rTrrrrcss|]}t|tjГVqdSr@)rOr"rPйr┌trrrrEВrz/test_tensorflow_with_sharing..)rr rrr"r#r$r%r┌shared_intermediates┌get_sharing_cache┌lenrM┌valuesr!r') rr(r)r+r,r-┌cacheZtfl1┌cache_szZtfl2rrr┌test_tensorflow_with_sharingos Lr`zTheano not installed.cCsРtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бsdJВd dД|DГ}||О}t|tjj ГsМJВdS) NFr cSsg|] }|jСqSrrrrrrrНrztest_theano..rTr┌theanocSsg|]}tа|бСqSr)r┌ to_theanorrrrrХr) rr rrr!r'rOra┌tensor┌TensorVariable)rr(r)r+r,r*Ztheano_viewsZ theano_optrrr┌test_theanoИs reztheano 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ГГsаJВtа ||бs░JВ|Иd dН}tа ||бs╠JВ|t аИбГ}t|t jjГsьJВdS)Nr2r3r9cs(g|] \}}|Иvr tjj|Оn|СqSrr:r=r?rrrаrz.test_theano_with_constants..c3s"|]}|ИvrИ|nИVqdSr@rrArBrrrEвrz-test_theano_with_constants..r5r/rarFcss"|]}|dupt|ГdkVqdS)NrarGrHrrrrEиrrJ)rKr!r;r<rrLrrMrNr'rrbrOrarcrdrQrrBr┌test_theano_with_constantsЪs&rfc Cs№tа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}tабП`}||ddiО}tаб|usnJВt|Г}|d ksВJВ||ddiО}t|Г|ksаJВWdГn1s┤0Ytd dД|а бDГГs╪JВt а||бsшJВt а||бs°JВdS)NFr cSsg|] }|jСqSrrrrrrr║rz,test_theano_with_sharing..rTrrarcss|]}t|tjjГVqdSr@)rOrarcrdrXrrrrE┼rz+test_theano_with_sharing..)rr rrrrZr[r\rMr]r!r') rr(r)r+r,r^Zthn1r_Zthn2rrr┌test_theano_with_sharing┤s .rgzCupy not installed.cCsдtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бsdJВd dД|DГ}||О}t|tjГsКJВtа|tа |ббsаJВdS) NFr cSsg|] }|jСqSrrrrrrr╨rztest_cupy..rTr┌cupycSsg|]}tа|бСqSr)rZto_cupyrrrrr╪r) rr rrr!r'rOrh┌ndarrayZasnumpy)rr(r)r+r,r*Z cupy_viewsZcupy_optrrr┌ test_cupy╦s rjc sd}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsаJВtа ||бs░JВ|Иd dН}tа ||бs╠JВ|t аИбГ}t|t j ГsъJВtа ||аббs■JВdS)Nr2r3r9cs(g|] \}}|Иvr tjj|Оn|СqSrr:r=r?rrrфrz,test_cupy_with_constants..c3s"|]}|ИvrИ|nИVqdSr@rrArBrrrEцrz+test_cupy_with_constants..r5r/rhrFcss"|]}|dupt|ГdkVqdS)NrhrGrHrrrrEэrrJ)rKr!r;r<rrLrrMrNr'rhZasarrayrOri┌getrQrrBr┌test_cupy_with_constants▐s&rlzjax not installed.cCsxtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бsdJВt|tjГstJВdS) NFr cSsg|] }|jСqSrrrrrrr rztest_jax..rTr┌jax)rr rrr!r'rOri)rr(r)r+r,r*rrr┌test_jax·s rncs┤d}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsаJВtа ||бs░JВdS) Nr2r3r9cs(g|] \}}|Иvr tjj|Оn|СqSrr:r=r?rrrrz+test_jax_with_constants..c3s"|]}|ИvrИ|nИVqdSr@rrArBrrrErz*test_jax_with_constants..r5r/rmrFcss"|]}|dupt|ГdkVqdS)NrmrGrHrrrrEr) rKr!r;r<rrLrrMrNr')r/rRrSrTrUr,rVrrBr┌test_jax_with_constantss&roc s╞d}d}ddД|DГ}t|g|вRОЙИ|О}tаИб}||Оаб}|tj|ddНksZJВtаtаЗfddДбб}||Г}td d Дt||ГDГГsЦJВddДt||ГDГ}||Оаб} | |ks┬JВdS)N· ij,jk,kl->йr4r6)r7r1cSsg|]}tjj|ОСqSrйr!r;Zrandnйr┌srrrr rz)test_jax_jit_gradient..gёhуИ╡°ф>)┌relcsИ|ОSr@rйr(йr,rr┌)rz'test_jax_jit_gradient..css|]\}}|j|jkVqdSr@rйr┌v1┌v2rrrrE+rz(test_jax_jit_gradient..cSsg|]\}}|d|СqSйg№йё╥MbP?rйrrZdvrrrr.r) rrmZjit┌item┌pytestZapprox┌gradrM┌zip) rRrSr(┌x0Zjit_expr┌x1┌ grad_expr┌ view_grads┌ new_viewsZx2rrwr┌test_jax_jit_gradients rЗzautograd not installed.csРd}d}ddД|DГ}t|g|вRОЙИ|О}tаЗfddДб}||Г}tddДt||ГDГГsdJВd dДt||ГDГ}И|О}||ksМJВdS) NrprqcSsg|]}tjj|ОСqSrrrrsrrrr7rz*test_autograd_gradient..csИ|ОSr@rrvrwrrrx<rz(test_autograd_gradient..css|]\}}|j|jkVqdSr@rryrrrrE>rz)test_autograd_gradient..cSsg|]\}}|d|СqSr|rr}rrrrAr)r┌autogradrАrMrБ)rRrSr(rВrДrЕrЖrГrrwr┌test_autograd_gradient3srЙcs╩tаdбЙtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}ЗfddД|DГ}||О}t|ИjГszJВtа |tа |ббsРJВt|g|вRО}t|ИjГs░JВtа |tа |ббs╞JВdS) Nz dask.arrayFr cSsg|] }|jСqSrrrrrrrLrztest_dask..rTcsg|]}Иj|ddНСqS)r1)┌chunks)Z from_arrayйr┌xй┌darrrPr)r┌importorskiprr rrrO┌Arrayr!r'rI)rr(r)r+r,Zda_viewsZda_optrrНr┌ test_daskFs rСc stаdбЙtа|б}|D]4}tjаdбtjаddg|jdddgб}d||<qt |g|вRdddЬО}d d Д|DГ}t |g|вRddiО}Зfdd Д|DГ}||О}t|ИjГs┤JВtа ||аббs╚JВt |g|вRО}t|ИjГsшJВtа ||аббs№JВdS) N┌sparseщ*FTgЪЩЩЩЩЩй?gffffffю?rr cSsg|] }|jСqSrrrrrrrkrztest_sparse..rcsg|]}Иjа|бСqSr)┌COOZ from_numpyrЛйrТrrror)rrПrr r!r;┌seed┌choicerrrrOrФr'Ztodense) rr(r┌maskr)r+r,Zsparse_viewsZ sparse_optrrХr┌test_sparse^s rЩzTorch not installed.cCsжtа|б}t|g|вRdddЬО}ddД|DГ}t|g|вRddiО}||ddiО}tа||бsdJВd dД|DГ}||О}t|tjГsКJВtа||а ба ббsвJВdS) NFr cSsg|] }|jСqSrrrrrrrГrztest_torch..rTr┌torchcSsg|]}tа|бСqSr)r┌to_torchrrrrrЛr)rr rrr!r'rOrЪrP┌cpurJ)rr(r)r+r,r*Ztorch_viewsZ torch_optrrr┌ test_torch}s rЭc s d}d}hdгИ\}ЗfddДt|ГDГЙtjj||ОЙt|gЗЗЗfddДtdГDГвRО}t|gИвRd ИiО}|Иd dН}tddД|jd DГГsаJВtа ||бs░JВ|Иd dН}tа ||бs╠JВ|t аИбГ}t|t jГsъJВ|jjdkr■|абn |абаб}tа ||бРsJВdS)Nr2r3r9cs(g|] \}}|Иvr tjj|Оn|СqSrr:r=r?rrrЧrz-test_torch_with_constants..c3s"|]}|ИvrИ|nИVqdSr@rrArBrrrEЩrz,test_torch_with_constants..r5r/rЪrFcss"|]}|dupt|ГdkVqdS)NrЪrGrHrrrrEЯrrJrЬ)rKr!r;r<rrLrrMrNr'rrЫrOrЪrPZdevice┌typerJrЬrQrrBr┌test_torch_with_constantsСs & rЯcCs0tdГ}t|ГdksJВt|gdГdks,JВdS)N)r0r1r5┌ opt_einsum┌autorJr)rМrrr┌+test_auto_backend_custom_array_no_tensordotнsrвcCs╥tа|б}t|g|вRdddЬО}|jtks0JВddД|DГ}t|g|вRddiО}ddД|DГ}t|g|вRdd iО}|jtksЖJВtа||аt ббsЬJВ||dd iО}|jtks╕JВtа||аt ббs╬JВdS) NFr cSsg|] }|jСqSrrrrrrr║rz.test_object_arrays_backend..rTcSsg|]}|аtбСqSr)┌astype┌objectrrrrr╜rrrд) rr rZdtyperдrr!r'rг┌float)rr(r)r+r,Z obj_viewsZobj_optrrr┌test_object_arrays_backend┤s rж)7rJr!rrаrrrrrZopt_einsum.contractrr r rhZ found_cupy┌ImportErrorrr"ZConfigProtor$Zgpu_optionsZallow_growthZfound_tensorflow┌os┌environraZfound_theanorЪZfound_torchrmZ found_jaxrИZfound_autograd┌tests┌markZskipifZparametrizer.rWr`rerfrgrjrlrnrorЗrЙrСrЩrЭrЯrвrжrrrr┌sк