tf81ddlmZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZddlmZddlmZddlmZdd lmZmZe j:eej>fej@ej:eej>fejBe j:eej>fejDe j:eej>fejFe j:ejHjJejHjFe j:eej>fee j:ejHjJeej:ej>d%d Z&e j:ejHjJd&d Z'ej:ejHjJd'd Z(ejRd e jRd ejRd ejRd ejRd dZ*ejRde jRddZ+ejRde jRdejRdejRddZ,ejRde jRddZ-dZ.ej:ejHjJdZ/ej:eej>fdZ0dZ1d(dZ2ej:eej>fdZ3dZ4GddeZ5ej:ej>dZ6Gdd e5Z7ed!d"e5d#$Z8e8jsd"e7y))) annotationsN)chunk)Array) concatenate_lookup divide_lookup einsum_lookup empty_lookupnannumel_lookup numel_lookuppercentile_lookuptensordot_lookupto_cupy_dispatchto_numpy_dispatch)divide) ma_divide _percentile)CreationDispatchDaskBackendEntrypointct|||SNr)aqmethods ]/var/www/html/software/conda/envs/higlass/lib/python3.12/site-packages/dask/array/backends.py percentiler#s q!V $$c6tjj||}|Dcgc]}t|ds|j}}t d|Dr t d|r-tj|}t|dk(r |d|_|Scc}w)N)axis fill_valuec3PK|]}t|tj ywr) isinstancenpndarray).0fs r z_concatenate..,s ::a $ :s$&zADask doesn't support masked array's with non-scalar `fill_value`sr) r#ma concatenatehasattrr any ValueErroruniquelen)arraysrouti fill_valuess r _concatenater4(s %%  F  .C)/LA71l3K1<<LKL :k :: O  ii , { q (^CN JMs BBc t||\}} t |}t|} t |}t|}|j }|j }|j } |j } d} ||k7rd} nPt|D]B} ||| | || k7rd} n-|| dkr || xx|z cc<|| dks6|| xx| z cc<D| s tdt|D cgc] } | |vs|  } } | |z}d}|D] }|||z} d|f}| Dcgc]}|| }}t| D cgc] } | |vs|  } } || z}d}|D] }|| |z} |df}| Dcgc]}| | }}|j|j|}|j|j|}tjj||}|j||zS#t$r/tt| d}ttd|}YwxYw#t$r |g}d}YwxYw#t$r |g}d}YwxYwcc} wcc}wcc} wcc}w)Nrr(TFzshape-mismatch for sum) iter TypeErrorlistranger/shapendimr- transposereshaper#r)dot)rbaxesaxes_aaxes_bnanbas_ndabsndbequalknotin newaxes_aN2r newshape_aolda newaxes_b newshape_boldbatbtress r _tensordotrW8s T  [f [f ''C &&C B &&C E Rxr !A6!9~F1I.ay1}q S ay1}q S  ! 122c 61avoQ 6E 6I B c$ibJ"' ($CI (D (c 61avoQ 6E 6I B bhbJ!& 'BtH 'D ' Y  ' ' 3B Y  ' ' 3B %%))B C ;;td{ ##w &eTE1o&eAtn%&    4 7 ) 6 (sW GHH-$ I.I I- I 7I  I4HHH*)H*-H?>H?cupycddltjjjt jjj tjjttjjttjjttjjfd}tjtjfd}t!jjfd}y)Nrc*j|fi|Sr)asnumpydatakwargsrXs r cupy_to_numpyz$register_cupy..cupy_to_numpyt||D+F++rc*j|fi|Sr)asarrayr\s r numpy_to_cupyz$register_cupy..numpy_to_cupyr`rcp|jdd|jddj|i|S)Ncastingorder)popeinsum)argsr^rXs r _cupy_einsumz#register_cupy.._cupy_einsums7  9d# 7D!t{{D+F++r)rXrregisterr$r*r tensordotr rr _numel_arrayliker _nannumelrrr#r)r_rcrjrXs @r register_cupyro|s  d.>.>?dllDNN;t||Z8$,,(89T\\95 -,.,rzz*,+,DLL),*,rcupyxcddlm} ddlmmdfd }t j ||tj |ty#t$r}t d|d}~wwxYw)Nr)spmatrix)hstackvstackz>Stacking of sparse arrays requires at least CuPy version 8.0.0cX|dk(r|S|dk(r|Sd|z}t|)Nrr(zECan only concatenate cupy sparse matrices for axis in {0, 1}. Got %s)r-)Lrmsgrsrts r_concat_cupy_sparsez+register_cupyx.._concat_cupy_sparsesC 19!9  QY!9 "$() S/ !rr) cupyx.scipy.sparserrrsrt ImportErrorrrkr _tensordot_scipy_sparse)rrerxrsrts @@rregister_cupyxr~s^,5 "*=>h(?@%  L  sA A$ AA$sparsecDddl}tj|j|jt j|j|j tj|jttj|jty)Nr) rrrkCOOr*r rlr _numel_ndarrayr _nannumel_sparse)rs rregister_sparsersf  F,>,>?fjj&*:*:;&**n5VZZ)9:rscipycddldfd }tjjj|t jjjt y)Nrc|dk(rjj|S|dk(rjj|Sd|z}t|)Nrr(zFCan only concatenate scipy sparse matrices for axis in {0, 1}. Got %s)rrtrsr-)rvrrwrs rr4z+register_scipy_sparse.._concatenatesY 19<<&&q) ) QY<<&&q) )"$() S/ !rry) scipy.sparserrkrrrr r|)r4rs @rregister_scipy_sparsersD " 5 5|Dell335LMrc|j|jcxk(rdk(sJJt|dt|dcxk(rdk(sJJ|d\}|d\}|dvr|dvsJ|j||j|k(sJ|dk(r|dk(r|j|zS|dk(r|dk(r|j|jzS|dk(r |dk(r||zS|dk(r|dk(r||jzSyy)Nrr()rr()r<r/r;T)rr@rAa_axisb_axiss rr|r|s 66QVV q   tAw<3tAw< ,1 ,, ,, ,QIVQIV V & 00 0 776?aggfo -- - {v{ssQw 11ssQSSy 111u 11133w%rc Ttjtj|fi|S)z'Numel implementation for masked arrays.)rsumr# ones_likexr^s r _numel_maskedrs  99R\\!_ / //rc t|fddi|S)zJNumel implementation for arrays that want to return numel of type ndarray.coerce_np_ndarrayT_numelrs rrrs ! 6t 6v 66rc t|fddi|S)zKNumel implementation for arrays that want to return numel of the same type.rFrrs rrmrms ! 7u 7 77rc |j|jdd}|jdd|jdtj}htj|}|dur|S|r$tj dt z||Stj||dt z| Stttfsgtjfd D}|d ur(tfd tt D}n'tfd tt D}|r+tjtj|||Stj|||| S)a  A reduction to count the number of elements. This has an additional kwarg in coerce_np_ndarray, which determines whether to ensure that the resulting array is a numpy.ndarray, or whether we allow it to be other array types via `np.full_like`. keepdimsFrNdtype)r)r()r;r r)r;rc3(K|] }| ywr)r%dimr;s rr'z_numel..s0CU3Z0sTc34K|]}|vr|ndyw)r(Nrr%rrr;s rr'z_numel..s% 58#T/E#Jq 0 sc32K|]}|vs|ywrrrs rr'z_numel..sUSPT_%*Us  )r;getr#float64prodfullr/ full_liker"tupler9mathr: broadcast_toarray) rrr^rrr new_shaperr;s @@rrrsA GGEzz*e,H ::fd #D JJw +E |wwuE* u K 77E !2t5Q Q<<4tc%j/@N N dUDM *v 99040 0D4 rs" !   8--AVRZZ0"..A62::. =  +RYY7 vrzz*BHH5 bee(("%%++6  +Y7 ruu))95BJJ'%(%RUU//0 1 255--.@$/@$F '!!!&)v&F#   (,)$'*( ,2 (!!!'*A+)A4 )!!!(+x(H% ;&),* ; (!!!'*N+)N&"ruu))*0+0  +,7-7 8 %CP62::.//0/ 7@"2@"FBJJ'(3*+ + " ((2H2JKr