`i3ddlmZddlmZmZmZmZmZerddlm Z m Z m Z m Z m Z ddlmZddlmZddlZddlm Zdd lmZd Zdddd dCdZ dDddddEdZddddFd ZddddGd"Z dHdddd#dId)ZdJd+ZddddKd-ZddddLd.Z ddd/d0dMd4Z!d5d6dNd;Z"ddddFd<Z#ddddGd=Z$dd>dOd?Z%dd>dOd@Z&ddddFdAZ'ddddGdBZ(dS)P) annotations) TYPE_CHECKINGListOptionalTupleUnion)ArrayDeviceDtypeNestedSequenceSupportsBufferProtocol)Sequence) _all_dtypesN)r )runtimecHdtzD] }||urdS td)NNz)dtype must be one of the supported dtypes)r ValueError)dtypeds v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/array_api/_creation_functions.py_check_valid_dtypers< { " A:: FF  @ A AA)rdevicecopyobjZUnion[Array, bool, int, float, NestedSequence[bool | int | float], SupportsBufferProtocol]rOptional[Dtype]rOptional[Device]rOptional[bool]returnr c~ddlm}t||'t|tst d||t }|durt dt||r| |j|krd}|durtj } tj |j |j tj|jd|}tj |n#tj |wxYw|S|0t|t r|d ks|d krt#d tj } tj |j tj|| }tj |n#tj |wxYw|j |S) z~ Array API compatible wrapper for :py:func:`np.asarray `. See its docstring for more information. r r NUnsupported device Fz!copy=False is not yet implementedT)rrllz&Integer out of bounds for array dtypesr) _array_objectr r isinstance_DevicerNotImplementedErrorrr getDevice setDeviceid_newnparray_arrayint OverflowErrorasarray)rrrrr prev_deviceress rr3r3!s,%$$$$$u *VW"="=9v99::: ~ u}}!"EFFF#u   e!3!3D 4<<!+--K /!&),,, ej#*4u!M!M!MNN!+....!+....  }C--}3==C*DTDTDEEE#%%K'&)$$$jE***+&&&&+&&&& 5:c??sAC11D/FF,)rrstopOptional[Union[int, float]]stepUnion[int, float]startcddlm}t||'t|tst d||t }t j} t j|j |j tj ||||t j|S#t j|wxYw)z| Array API compatible wrapper for :py:func:`np.arange `. See its docstring for more information. r r#Nr$)r6r8r) r&r rr'r(rrr*r+r,r-r.arange)r:r6r8rrr r4s rr<r<Ys%$$$$$u *VW"="=9v99::: ~#%%K'&)$$$uz")E4uMMMNN+&&&&+&&&& #>B66C shapeUnion[int, Tuple[int, ...]]cddlm}t||'t|tst d||t }t j} t j|j |j tj ||t j|S#t j|wxYw)zz Array API compatible wrapper for :py:func:`np.empty `. See its docstring for more information. r r#Nr$r%) r&r rr'r(rrr*r+r,r-r.emptyr>rrr r4s rrArAv%$$$$$u *VW"="=9v99::: ~#%%K'&)$$$uz"(566677+&&&&+&&&& #`. See its docstring for more information. r r#Nr$r%)r&r rr'r(rrr*r+r,r-r. empty_liker0rErrr r4s rrGrG%$$$$$u *VW"="=9v99::: ~#%%K'&)$$$uz"->>>??+&&&&+&&&& #AB99C)krrn_rowsr1n_cols Optional[int]rKcddlm}t||'t|tst d||t }t j} t j|j |j tj ||||t j|S#t j|wxYw)zv Array API compatible wrapper for :py:func:`np.eye `. See its docstring for more information. r r#Nr$)MrKr) r&r rr'r(rrr*r+r,r-r.eye)rLrMrKrrr r4s rrQrQs%$$$$$u *VW"="=9v99::: ~#%%K'&)$$$uz"&6QeDDDEE+&&&&+&&&&r=objectcRddlm}|jtj|S)z Array API compatible wrapper for :py:func:`np.from_dlpack `. See its docstring for more information. r r#)r&r r-r. from_dlpack)rEr s rrTrTs2 %$$$$$ 5:bnQ'' ( ((r fill_valuec ddlm}t||'t|tst d||t }t||r|jdkr|j}tj } tj |j tj |||}tj |n#tj |wxYw|jtvrt!d|j|S)zx Array API compatible wrapper for :py:func:`np.full `. See its docstring for more information. r r#Nr$rr%zInvalid input to full)r&r rr'r(rndimr0rr*r+r,r.fullrr TypeErrorr-)r>rUrrr r4r5s rrXrXs%$$$$$u *VW"="=9v99::: ~*e$$'A)=)=& #%%K'&)$$$geZu555+&&&&+&&&& y ##/000 5:c??s 0C C c*ddlm}t||'t|tst d||t }t||r|jdkr|j}tj } tj |j tj |j||}tj |n#tj |wxYw|jtvrt!d|j|S)z Array API compatible wrapper for :py:func:`np.full_like `. See its docstring for more information. r r#Nr$rr%zInvalid input to full_like)r&r rr'r(rrWr0rr*r+r,r. full_likerrrYr-)rErUrrr r4r5s rr[r[s%$$$$$u *VW"="=9v99::: ~*e$$'A)=)=& #%%K'&)$$$l18Zu===+&&&&+&&&& y ##4555 5:c??s 5CC%T)rrendpointnumr\boolc ddlm}t||t}n't |tst d|t j} t j|j |j tj |||||t j|S#t j|wxYw)z Array API compatible wrapper for :py:func:`np.linspace `. See its docstring for more information. r r#Nr$)rr\) r&r rr(r'rrr*r+r,r-r.linspace)r:r6r]rrr\r r4s rr`r`s%$$$$$u ~  ( (;9v99:::#%%K'&)$$$uz"+eT3ehWWWXX+&&&&+&&&&s "?B66C xy)indexingarraysrbstr List[Array]cddlmtd|Ddkrtdfdt jd|Dd|iDS)z Array API compatible wrapper for :py:func:`np.meshgrid `. See its docstring for more information. r r#ch|] }|j Sr%.0as r zmeshgrid..<s $ $ $AG $ $ $rz,meshgrid inputs must all have the same dtypec0g|]}j|Srh)r-)rjr/r s r zmeshgrid..?s5      5   rcg|] }|j Srh)r0ris rrnzmeshgrid..As"<"<"<18"<"<"`. See its docstring for more information. r r#Nr$r%) r&r rr'r(rrr*r+r,r-r.onesrBs rrsrsEs%$$$$$u *VW"="=9v99::: ~#%%K'&)$$$uz"'%u55566+&&&&+&&&&rDcddlm}t||'t|tst d||t }t j} t j|j |j tj |j |t j|S#t j|wxYw)z Array API compatible wrapper for :py:func:`np.ones_like `. See its docstring for more information. r r#Nr$r%)r&r rr'r(rrr*r+r,r-r. ones_liker0rHs rruru_s%$$$$$u *VW"="=9v99::: ~#%%K'&)$$$uz",qxu===>>+&&&&+&&&&rJrKcddlm}|jdkrtd|jt j|j|S)zx Array API compatible wrapper for :py:func:`np.tril `. See its docstring for more information. r r#z)x must be at least 2-dimensional for trilrv)r&r rWrr-r.trilr0rErKr s rryryvS %$$$$$vzzDEEE 5:bgah!,,, - --rcddlm}|jdkrtd|jt j|j|S)zx Array API compatible wrapper for :py:func:`np.triu `. See its docstring for more information. r r#rxz)x must be at least 2-dimensional for triurv)r&r rWrr-r.triur0rzs rr}r}r{rcddlm}t||'t|tst d||t }t j} t j|j |j tj ||t j|S#t j|wxYw)zz Array API compatible wrapper for :py:func:`np.zeros `. See its docstring for more information. r r#Nr$r%) r&r rr'r(rrr*r+r,r-r.zerosrBs rrrrCrDcddlm}t||'t|tst d||t }t j} t j|j |j tj |j |t j|S#t j|wxYw)z Array API compatible wrapper for :py:func:`np.zeros_like `. See its docstring for more information. r r#Nr$r%)r&r rr'r(rrr*r+r,r-r. zeros_liker0rHs rrrrIrJ) rrrrrrrr r!r )Nr ) r6r7r8r9r:r9rrrrr!r )r>r?rrrrr!r )rEr rrrrr!r r) rLr1rMrNrKr1rrrrr!r )rErRr!r ) r>r?rUr9rrrrr!r ) rUr9rEr rrrrr!r )r]r1r:r9r6r9rrrrr\r^r!r )rcr rbrdr!re)rEr rKr1r!r )) __future__rtypingrrrrr_typingr r r r rcollections.abcr_dtypesrcupyr. cupy.cudar(cupy_backends.cuda.apirrr3r<rArGrQrTrXr[r`rqrsruryr}rrrhrrrsi""""""?>>>>>>>>>>>>>))((((( ''''''******BBB,"#555555v)- ' "#''''''@"# ''''''6.2d''''''2!' !#'''''':))))"# N"#      R"#''''''<.2."# ''''''6.2d''''''.#$ . . . . . .#$ . . . . . .""# ''''''6.2d''''''''r