o X۷it@sddlmZddlZddlZddlmZzddlmZ Wn e y'dZ YnwdZ iZ gZ ddZddZdd Zgd Ze rSeD]ZeeeZee eZee e<qBdSdS) ) annotationsNznumpy.scipy.linalgcCs\|rz dd|D}WntytYSwdd|D}tddt||Ds,tS|S)NcSs0g|]}|jr|jtjurt|jn|jqS) coercibletypenpndarraycupyasarrayvalue.0drrP/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/linalg/_uarray.py s z"__ua_convert__..cSsg|]}|jqSr)r r rrrrscss,|]\}}|jtjupt|tjVqdSN)rrr isinstancer)r rr rrr s z!__ua_convert__..) TypeErrorNotImplementedallzip) dispatchablescoercereplacedrrr__ua_convert__s rcCs&t|d}|dur tS||i|Sr) _implementedgetr)methodargskwargsfnrrr__ua_function__'s r"csfdd}|S)zBDecorator adds function to the dictionary of implemented functionscs.tottd}|r|t|<|St|Sr) _scipy_linalggetattrr _notfoundappend)func scipy_funcscipy_func_namerrinner0s zimplements..innerr)r*r+rr)r implements.s  r,) eigheigvalshcholeskyqrsvdnormdetsolvelstsqinvpinv) __future__rnumpyrr cupy.linalglinalg _cp_linalg scipy.linalgr# ImportError __ua_domain__rr%rr"r,_cp_linalg_functions func_namer$cp_funcr(rrrrs.