o "e@s@dZddlZddlmZgdZeddZdd Zd d ZdS) zK Required functions for optimized contractions of numpy arrays using cupy. N)to_backend_cache_wrap)to_cupybuild_expressionevaluate_constantscCs"ddl}t|tjr||S|S)Nr)cupy isinstancenpZndarrayZasarray)Zarrayrr 8lib/python3.10/site-packages/opt_einsum/backends/cupy.pyr s  rcsfdd}|S)zz;build_expression..cupy_contract..r)backend)Z _contractget)Zarraysexprr 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 )Z const_arraysrr r r rsr) __doc__Znumpyr Zsharingr__all__rrrr r r r s