
    `i                     r    d dl mZ d dl mZ d dlmZ d dlmZ ddZ ej        ddd	d
          Z	d Z
ddZdS )    )_kernel)_memory_range)join)searchNc                 N    |t          j        | |          S t          d          )Nz'Only supported for `max_work` is `None`)r   may_share_boundsNotImplementedError)abmax_works      l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_misc/memory_ranges.pymay_share_memoryr      s*    -a333
G
H
HH    zT xz
uint64 outzout = (unsigned long long)(&x)cupy_get_memory_ptrsc                     | j         j        dk    rt          |           S t          j        t          | j                  t          | j                  g          S )Nc)dtypekind_get_memory_ptrs_kernelr   concatenaterealimag)xs    r   _get_memory_ptrsr      sS    w|s&q)))''''   r   c                    | |u r| j         dk    rdS |dk    rt          j        | |          S |dv rt          |                                           }t          |                              dd          }|                                 t          j        ||d          }t          j        ||d          }t          ||k    
                                          S t          d	          )
Nr   TMAY_SHARE_BOUNDS)NMAY_SHARE_EXACT   leftrightz%Not supported for integer `max_work`.)sizer   r   r   ravelreshapesortr   searchsortedboolanyr	   )r
   r   r   a_ptrsb_ptrsr   ys          r   shares_memoryr,      s    Avv!&A++t%%%-a333,,,!!$$**,,!!$$,,R337788Q!VLLNN###
E
F
FFr   )N)
cupy._corer   r   cupy._manipulationr   cupy._sortingr   r   ElementwiseKernelr   r   r,    r   r   <module>r2      s          $ $ $ $ $ $ # # # # # #            I I I I 4'3	<$    G G G G G Gr   