o 2h@sDdZddlmZddlmZgdZeddZddZd d Zd S) zIRequired functions for optimized contractions of numpy arrays using cupy.)has_array_interface)to_backend_cache_wrap)to_cupybuild_expressionevaluate_constantscCsddl}t|r ||S|S)Nr)cupyrasarray)arrayrr S/var/www/html/chatgem/venv/lib/python3.10/site-packages/opt_einsum/backends/cupy.pyr s rcsfdd}|S)z7Build a cupy function based on ``arrays`` and ``expr``.csjdd|DddS)NcSg|]}t|qSr r.0xr r r z;build_expression..cupy_contract..r)backend) _contractget)arraysexprr r cupy_contractsz'build_expression..cupy_contractr )_rrr rr rs rcCs|dd|DdddS)zcConvert constant arguments to cupy arrays, and perform any possible constant contractions. cSr r r rr r r r rz&evaluate_constants..rT)rrr ) const_arraysrr r r rsrN) __doc__opt_einsum.helpersropt_einsum.sharingr__all__rrrr r r r s