B ▓Ї`Ї;у@s4ddlZddlZddlmZmZmZmZmZddl m Z mZmZyddl Z dZWnek rhdZYnXy ddlZeабZdej_dZWnek rвdZYnXy"ddlZdejd<ddlZdZWnek r▐dZYnXyddlZdZWnek Рr dZYnXyddlZdZWnek Рr6dZYnXyddlZdZ Wnek РrbdZ YnXdd d ddd ddgZ!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#eddНej"а$de!б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Нej"а$de!бd*d+ДГГZ-ej"j#ed)dНej"а$dddhddhddhgбd,d-ДГГZ.ej"j#ed)dНd.d/ДГZ/ej"j#e d0dНd1d2ДГZ0ej"а$de!бd3d4ДГZ1ej"а$de!бd5d6ДГZ2ej"j#ed7dНej"а$de!бd8d9ДГГZ3ej"j#ed7dНej"а$dddhddhddhgбd:d;ДГГZ4dd?ДГZ6dS)@щN)┌backends┌contract┌contract_expression┌helpers┌sharing)┌Shaped┌ infer_backend┌ parse_backendTFZGNUZMKL_THREADING_LAYERz ab,bc->cazabc,bcd,deazabc,def->fedcbazabc,bcd,df->fazijk,ikjzi,j->ijz ijk,k->ijz AB,BC->CAzTensorflow not installed.)┌reason┌stringc Csоtа|б}t|f|ЮdddЬО}tа|б}ddД|DГ}t|f|ЮddiО}tjtdН}|а бП||d|d ЬОWdQRX|а бtа||бsФtВd dД|DГ}||ОdS)NF)┌optimize┌use_blascSsg|] }|jСqSй)┌shape)┌.0┌vrr·M/home/dcms/DCMS/lib/python3.7/site-packages/opt_einsum/tests/test_backends.py· Dsz#test_tensorflow..rT)┌config┌ tensorflow)┌backend┌outcSsg|]}tа|бСqSr)r┌ to_tensorflow)r┌viewrrrrOs) r┌build_viewsr┌npZ empty_liker┌tf┌Session┌ _TF_CONFIG┌ as_default┌close┌allclose┌AssertionError)r┌views┌ein┌opt┌shps┌expr┌sessZtensorflow_viewsrrr┌test_tensorflow=s r)┌ constantsщщc s d}d}dddhИ\}ЗfddДt|ГDГЙtjj||ОЙt|fЗЗЗfdd Дtd ГDГЮО}t|fИЮdИiО}tjt dНа бП|Иd dН}WdQRXtdd Д|jd DГГs║t Вtа||бs╩t В|ИddН}tа||бsцt В|tаИбГ}t|tjГРst ВdS)Nzij,jk,kl->li))r,щ)r-щ)r.щrr+r,cs(g|] \}}|Иkr tjj|Оn|СqSr)r┌random┌rand)r┌i┌shp)r*rrrYsz2test_tensorflow_with_constants..c3s"|]}|ИkrИ|nИVqdS)Nr)rr2)r*┌ops┌varrr· [sz1test_tensorflow_with_constants..r-r*)rr)rcss"|]}|dkpt|ГdkVqdS)Nr)r)r┌arrayrrrr6bs┌numpy)┌ enumeraterr0r1r┌rangerrrrr┌all┌_evaluated_constantsr"r!rr┌ isinstance┌Tensor) r*┌eq┌shapes┌ non_const┌res_expr'┌res_got┌res_got2┌res_got3r)r*r4r5r┌test_tensorflow_with_constantsSs $rFc Cstа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}tjtdН}|абПjtа бПV}||dd iО}tа б|ksАtВt|Г}|d ksФtВ||dd iО} t|Г|ks▓tВWdQRXWdQRXt ddД|абDГГsрtВtа||бsЁtВtа|| бРstВdS) NF)rr cSsg|] }|jСqSr)r)rrrrrrusz0test_tensorflow_with_sharing..rT)rrrrcss|]}t|tjГVqdS)N)r=rr>)r┌trrrr6Вsz/test_tensorflow_with_sharing..)rrrrrrrrr┌shared_intermediates┌get_sharing_cacher"┌lenr;┌valuesrr!) rr#r$r&r'r(┌cacheZtfl1┌cache_szZtfl2rrr┌test_tensorflow_with_sharingos $rNzTheano not installed.cCsМtа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}||ddiО}tа||бs`tВd dД|DГ}||О}t|tj j ГsИtВdS) NF)rr cSsg|] }|jСqSr)r)rrrrrrНsztest_theano..rTr┌theanocSsg|]}tа|бСqSr)r┌ to_theano)rrrrrrХs)rrrrrr!r"r=rO┌tensor┌TensorVariable)rr#r$r&r'r%Ztheano_viewsZ theano_optrrr┌test_theanoИs rSztheano not installed.c sюd}d}dddhИ\}ЗfddДt|ГDГЙtjj||ОЙt|fЗЗЗfdd Дtd ГDГЮО}t|fИЮdИiО}|Иdd Н}tdd Д|jdDГГsЮt Вtа ||бsоt В|Иdd Н}tа ||бs╩t В|tаИбГ}t |tjjГsъt ВdS)Nzij,jk,kl->li))r,r-)r-r.)r.r/rr+r,cs(g|] \}}|Иkr tjj|Оn|СqSr)rr0r1)rr2r3)r*rrrаsz.test_theano_with_constants..c3s"|]}|ИkrИ|nИVqdS)Nr)rr2)r*r4r5rrr6вsz-test_theano_with_constants..r-r*rO)rcss"|]}|dkpt|ГdkVqdS)NrO)r)rr7rrrr6иsr8)r9rr0r1rr:rr;r<r"r!rrPr=rOrQrR) r*r?r@rArBr'rCrDrEr)r*r4r5r┌test_theano_with_constantsЪs$rTc Csфtа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}tабПV}||ddiО}tаб|ksjtВt|Г}|d ks~tВ||ddiО}t|Г|ksЬtВWdQRXt d dД|а бDГГs└tВtа||бs╨tВtа||бsрtВdS)NF)rr cSsg|] }|jСqSr)r)rrrrrr║sz,test_theano_with_sharing..rTrrOrcss|]}t|tjjГVqdS)N)r=rOrQrR)rrGrrrr6┼sz+test_theano_with_sharing..) rrrrrrHrIr"rJr;rKrr!) rr#r$r&r'rLZthn1rMZthn2rrr┌test_theano_with_sharing┤s rUzCupy not installed.cCsаtа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}||ddiО}tа||бs`tВd dД|DГ}||О}t|tj ГsЖtВtа|tа |ббsЬtВdS) NF)rr cSsg|] }|jСqSr)r)rrrrrr╨sztest_cupy..rTr┌cupycSsg|]}tа|бСqSr)rZto_cupy)rrrrrr╪s)rrrrrr!r"r=rV┌ndarrayZasnumpy)rr#r$r&r'r%Z cupy_viewsZcupy_optrrr┌ test_cupy╦s rXc sd}d}dddhИ\}ЗfddДt|ГDГЙtjj||ОЙt|fЗЗЗfdd Дtd ГDГЮО}t|fИЮdИiО}|Иdd Н}tdd Д|jdDГГsЮt Вtа ||бsоt В|Иdd Н}tа ||бs╩t В|tаИбГ}t |tjГsшt Вtа ||аббs№t ВdS)Nzij,jk,kl->li))r,r-)r-r.)r.r/rr+r,cs(g|] \}}|Иkr tjj|Оn|СqSr)rr0r1)rr2r3)r*rrrфsz,test_cupy_with_constants..c3s"|]}|ИkrИ|nИVqdS)Nr)rr2)r*r4r5rrr6цsz+test_cupy_with_constants..r-r*rV)rcss"|]}|dkpt|ГdkVqdS)NrV)r)rr7rrrr6эsr8)r9rr0r1rr:rr;r<r"r!rVZasarrayr=rW┌get) r*r?r@rArBr'rCrDrEr)r*r4r5r┌test_cupy_with_constants▐s$rZzjax not installed.cCsttа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}||ddiО}tа||бs`tВt|tjГsptВdS) NF)rr cSsg|] }|jСqSr)r)rrrrrr sztest_jax..rTr┌jax) rrrrrr!r"r=rW)rr#r$r&r'r%rrr┌test_jax·s r\cs▓d}d}dddhИ\}ЗfddДt|ГDГЙtjj||ОЙt|fЗЗЗfdd Дtd ГDГЮО}t|fИЮdИiО}|Иdd Н}tdd Д|jdDГГsЮt Вtа ||бsоt ВdS)Nzij,jk,kl->li))r,r-)r-r.)r.r/rr+r,cs(g|] \}}|Иkr tjj|Оn|СqSr)rr0r1)rr2r3)r*rrrsz+test_jax_with_constants..c3s"|]}|ИkrИ|nИVqdS)Nr)rr2)r*r4r5rrr6sz*test_jax_with_constants..r-r*r[)rcss"|]}|dkpt|ГdkVqdS)Nr[)r)rr7rrrr6s)r9rr0r1rr:rr;r<r"r!)r*r?r@rArBr'rCr)r*r4r5r┌test_jax_with_constantss$r]c s─d}d}ddД|DГ}t|f|ЮОЙИ|О}tаИб}||Оаб}|tj|ddНksXtВtаtаЗfddДбб}||Г}td d Дt ||ГDГГsФtВddДt ||ГDГ}||Оаб} | |ks└tВdS)Nz ij,jk,kl->))r,r-)r-r.)r.r,cSsg|]}tjj|ОСqSr)rr0┌randn)r┌srrrr sz)test_jax_jit_gradient..gёhуИ╡°ф>)┌relcsИ|ОS)Nr)r#)r'rr┌)єz'test_jax_jit_gradient..css|]\}}|j|jkVqdS)N)r)r┌v1┌v2rrrr6+sz(test_jax_jit_gradient..cSsg|]\}}|d|СqS)g№йё╥MbP?r)rr┌dvrrrr.s) rr[Zjit┌item┌pytestZapproxr"┌gradr;┌zip) r?r@r#┌x0Zjit_expr┌x1┌ grad_expr┌ view_grads┌ new_viewsZx2r)r'r┌test_jax_jit_gradients rozautograd not installed.csОd}d}ddД|DГ}t|f|ЮОЙИ|О}tаЗfddДб}||Г}tddДt||ГDГГsbtВd dДt||ГDГ}И|О}||ksКtВdS) Nz ij,jk,kl->))r,r-)r-r.)r.r,cSsg|]}tjj|ОСqSr)rr0r^)rr_rrrr7sz*test_autograd_gradient..csИ|ОS)Nr)r#)r'rrra<rbz(test_autograd_gradient..css|]\}}|j|jkVqdS)N)r)rrcrdrrrr6>sz)test_autograd_gradient..cSsg|]\}}|d|СqS)g№йё╥MbP?r)rrrerrrrAs)r┌autogradrhr;rir")r?r@r#rjrlrmrnrkr)r'r┌test_autograd_gradient3srqcs─tаdбЙtа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}ЗfddД|DГ}||О}t|ИjГsvtВt а |t а|ббsМtВt|f|ЮО}t|ИjГsкtВt а |t а|ббs└tВdS) Nz dask.arrayF)rr cSsg|] }|jСqSr)r)rrrrrrLsztest_dask..rTcsg|]}Иj|ddНСqS)r,)┌chunks)Z from_array)r┌x)┌darrrPs)rg┌importorskiprrrrr=┌Arrayr"rr!r7)rr#r$r&r'Zda_viewsZda_optr)rtr┌ test_daskFs rwc s■tаdбЙtа|б}x<|D]4}tjаdбtjаddg|jdddgб}d||<qWt |f|ЮdddЬО}d d Д|DГ}t |f|ЮddiО}Зfdd Д|DГ}||О}t|ИjГs┤t Вtа||аббs╚t Вt |f|ЮО}t|ИjГsцt Вtа||аббs·t ВdS) N┌sparseщ*FTgЪЩЩЩЩЩй?gffffffю?r)rr cSsg|] }|jСqSr)r)rrrrrrksztest_sparse..rcsg|]}Иjа|бСqSr)┌COOZ from_numpy)rrs)rxrrros)rgrurrrr0┌seed┌choicerrrr=rzr"r!Ztodense) rr#r┌maskr$r&r'Zsparse_viewsZ sparse_optr)rxr┌test_sparse^s r~zTorch not installed.cCsвtа|б}t|f|ЮdddЬО}ddД|DГ}t|f|ЮddiО}||ddiО}tа||бs`tВd dД|DГ}||О}t|tj ГsЖtВtа||а баббsЮtВdS) NF)rr cSsg|] }|jСqSr)r)rrrrrrГsztest_torch..rTr┌torchcSsg|]}tа|бСqSr)r┌to_torch)rrrrrrЛs)rrrrrr!r"r=rr>┌cpur8)rr#r$r&r'r%Ztorch_viewsZ torch_optrrr┌ test_torch}s rВc sd}d}dddhИ\}ЗfddДt|ГDГЙtjj||ОЙt|fЗЗЗfdd Дtd ГDГЮО}t|fИЮdИiО}|Иdd Н}tdd Д|jdDГГsЮt Вtа ||бsоt В|Иdd Н}tа ||бs╩t В|tаИбГ}t |tjГsшt В|jjdkr№|абn |абаб}tа ||бРst ВdS)Nzij,jk,kl->li))r,r-)r-r.)r.r/rr+r,cs(g|] \}}|Иkr tjj|Оn|СqSr)rr0r1)rr2r3)r*rrrЧsz-test_torch_with_constants..c3s"|]}|ИkrИ|nИVqdS)Nr)rr2)r*r4r5rrr6Щsz,test_torch_with_constants..r-r*r)rcss"|]}|dkpt|ГdkVqdS)Nr)r)rr7rrrr6Яsr8rБ)r9rr0r1rr:rr;r<r"r!rrАr=rr>Zdevice┌typer8rБ) r*r?r@rArBr'rCrDrEr)r*r4r5r┌test_torch_with_constantsСs $ rДcCs0tdГ}t|ГdkstВt|gdГdks,tВdS)N)r+r,r-┌ opt_einsum┌autor8)rrr"r )rsrrr┌+test_auto_backend_custom_array_no_tensordotнsrЗcCs╠tа|б}t|f|ЮdddЬО}|jtks.tВddД|DГ}t|f|ЮddiО}ddД|DГ}t|f|Юdd iО}|jtksАtВtа||а t ббsЦtВ||dd iО}|jtks▓tВtа||а t ббs╚tВdS) NF)rr cSsg|] }|jСqSr)r)rrrrrr║sz.test_object_arrays_backend..rTcSsg|]}|аtбСqSr)┌astype┌object)rrrrrr╜srrЙ)rrrZdtyperЙr"rrr!rИ┌float)rr#r$r&r'Z obj_viewsZobj_optrrr┌test_object_arrays_backend┤s rЛ)7r8rrgrЕrrrrrZopt_einsum.contractrrr rVZ found_cupy┌ImportErrorrrZConfigProtorZgpu_optionsZallow_growthZfound_tensorflow┌os┌environrOZfound_theanorZfound_torchr[Z found_jaxrpZfound_autograd┌tests┌markZskipifZparametrizer)rFrNrSrTrUrXrZr\r]rorqrwr~rВrДrЗrЛrrrr┌sЦ *** **