
    >ieR                        d dl mZ d dlZd dlZej        j        Zd dlm	Z
 d dlmZ d dlmZ  ej        d          Zd ZdS )    )annotationsN)apply_gufunc)	assert_eqcupyc                    d } t           j                                                            d          }t	          j        |dd          }t          j        |d          }t          | d|dd	d
id          }t          ||           d S )Nc                *    t          j        |           S )N)npdiff)xs    Alib/python3.11/site-packages/dask/array/tests/test_cupy_gufunc.pymydiffz&test_apply_gufunc_axis.<locals>.mydiff   s    wqzz    )            F)chunksasarray   )axisz(i)->(i)i   T)r   output_sizesallow_rechunk)
r   randomdefault_rngstandard_normalda
from_arrayr	   r
   r   r   )r   ada_mdms        r   test_apply_gufunc_axisr$      s       	!!11)<<A
-!U
3
3
3C
A	
CasAhd
 
 
B ar   )
__future__r   numpyr	   pytestmarkgpu
pytestmark
dask.arrayarrayr   dask.array.gufuncr   dask.array.utilsr   importorskipr   r$    r   r   <module>r1      s    " " " " " "     [_
       * * * * * * & & & & & &v6""    r   