
    `i                     `   U d dl Z d dlmZmZmZ d dlZd dlZe j        d             Ze j        d             Z	e j        d             Z
e j        d             Z G d d          Zee         ZdZed         ed	<    ed
de	          Z edde          Z edde
          Z edde          ZdS )    N)CallableFinalOptionalc                     | j         dv r,|                     t          j        |           j                  S | j         dv r |                     t          j                   S t          d|            NbiufzUnsupported type: )kindtypenumpyiinfomininfRuntimeErrordtypes    r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/distributed/array/_modes.py_min_value_ofr   	   sh    zUzz%+e,,0111	s		zz59*%%%777888    c                     | j         dv r,|                     t          j        |           j                  S | j         dv r|                     t          j                  S t          d|            r   )r
   r   r   r   maxr   r   r   s    r   _max_value_ofr      sf    zUzz%+e,,0111	s		zz%)$$$777888r   c                 ,    |                      d          S )Nr   r   r   s    r   _zero_ofr          ::a==r   c                 ,    |                      d          S )N   r   r   s    r   _one_ofr   "   r   r   c                   x    e Zd ZU ej        j        j        ed<   ej        ed<   e	ed<   e
ed<   dede	de
ddfdZdS )	_OpModefunc
numpy_func
idempotentidentity_of	func_namereturnNc                     	 t          t          |          | _        t          t          |          | _        n# t
          $ r t          d          w xY w|| _        || _        d S )NzNo such function exists)	getattrcupyr"   r   r#   AttributeErrorr   r$   r%   )selfr&   r$   r%   s       r   __init__z_OpMode.__init__-   sl    	:i00DI%eY77DOO 	: 	: 	:8999	: %&s	   47 A)__name__
__module____qualname__r*   _core_kernelufunc__annotations__r   boolr   strr-    r   r   r!   r!   '   s         
*

""""
'
'*.
'=E
'	
' 
' 
' 
' 
' 
'r   r!   REPLICAminimumTmaximumaddFmultiply)	functoolstypingr   r   r   r   r*   	lru_cacher   r   r   r   r!   Moder8   r4   MINMAXSUMPRODr7   r   r   <module>rE      sZ        , , , , , , , , , ,   9 9 9 9 9 9      ' ' ' ' ' ' ' '&  t   gi$//gi$//geUH%%wz5'**r   