a SicDã@sdZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Zgd ¢Zd d d ddddœZd#dd„ZejejejejejejdœZd$dd„ZiZdd„ZiZdd„Ze je jejejejdœZe je jejejejdœZdd„Zdd „Zd!d"„Z dS)%z¦ Handles dispatching array operations to the correct backend library, as well as converting arrays to backend formats and then potentially storing them as constants. éNé)Ú object_arrays)Úcupy)Újax)Ú tensorflow)Útheano)Útorch)Úget_funcÚ has_einsumÚ has_tensordotÚbuild_expressionÚevaluate_constantsÚ has_backendz dask.arrayz theano.tensorzopt_einsum.backends.torchz jax.numpyzautograd.numpyz mars.tensor)ZdaskrrrÚautogradZmarscCs^z2t t ||¡¡}|dur&t||ƒn t|||ƒWStyXd}t| ||¡ƒ‚Yn0dS)z½Try and import ``{backend}.{func}``. If library is installed and func is found, return the func; otherwise if default is provided, return default; otherwise raise an error. Nz´{} doesn't seem to provide the function {} - see https://optimized-einsum.readthedocs.io/en/latest/backends.html for details on which functions are required for which contractions.)Ú importlibÚ import_moduleÚ_aliasesÚgetÚgetattrÚAttributeErrorÚformat)ÚfuncÚbackendÚdefaultÚlibÚ error_msg©rúX/var/www/html/django/DPS/env/lib/python3.9/site-packages/opt_einsum/backends/dispatch.pyÚ _import_funcs   r))Ú tensordotÚnumpy)Ú transposer )Úeinsumr )rÚobject)r!r#)r"r#r cCsBzt||fWSty<t|||ƒ}|t||f<|YS0dS)ziReturn ``{backend}.{func}``, e.g. ``numpy.einsum``, or a default func if provided. Cache result. N)Ú _cached_funcsÚKeyErrorr)rrrÚfnrrrr <s    r c Cs\z t|WStyVztd|ƒdt|<WntyHdt|<Yn0t|YS0dS)zHCheck if ``{backend}.einsum`` exists, cache result for performance. r"TFN)Ú _has_einsumr%r r©rrrrr Ls     r c Cs\z t|WStyVztd|ƒdt|<WntyHdt|<Yn0t|YS0dS)zKCheck if ``{backend}.tensordot`` exists, cache result for performance. rTFN)Ú_has_tensordotr%r rr(rrrr ^s     r )rrrrrcCst|||ƒS)zxBuild an expression, based on ``expr`` and initial arrays ``arrays``, that evaluates using backend ``backend``. )ÚCONVERT_BACKENDS©rÚarraysÚexprrrrr €sr cCst|||ƒS)zƒConvert constant arrays to the correct backend, and perform as much of the contraction of ``expr`` with these as possible. )ÚEVAL_CONSTS_BACKENDSr+rrrr ‡sr cCs | ¡tvS)z$Checks if the backend is known. )Úlowerr*r(rrrrŽsr)N)r N)!Ú__doc__rr ÚrrZ_cupyrZ_jaxrZ _tensorflowrZ_theanorZ_torchÚ__all__rrrr!r"Z object_einsumr$r r'r r)r r r*r r.rrrrrÚsX      ú ù û û