
    biR                        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y)    )annotationsN)apply_gufunc)	assert_eqcupyc                     d } t         j                  j                         j                  d      }t	        j
                  |dd      }t        j                  |d      }t        | d|dd	d
id      }t        ||       y )Nc                ,    t        j                  |       S )N)npdiff)xs    \/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/tests/test_cupy_gufunc.pymydiffz&test_apply_gufunc_axis.<locals>.mydiff   s    wwqz    )            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$      so     	!11)<A
--!U
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   