o W۷i8@sddlmZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddd Z dd d Z dd d ZdddZdddZ    dddZddZ    dddddZ    dddddZddZdS) ) annotationsN)_core)_routines_statistics)_fusion_thread_local)contentFcC8tr|r tdtjtj|||dS|j|||dS)aReturns the minimum of an array or the minimum along an axis. .. note:: When at least one element is NaN, the corresponding min value will be NaN. Args: a (cupy.ndarray): Array to take the minimum. axis (int): Along which axis to take the minimum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The minimum of ``a``, along the axis if specified. .. note:: When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements. .. seealso:: :func:`numpy.amin` z4cupy.amin does not support `keepdims` in fusion yet.axisoutr r keepdims)r is_fusingNotImplementedErrorcall_reduction _statisticsaminminar r r rL/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_statistics/order.pyr rcCr)aReturns the maximum of an array or the maximum along an axis. .. note:: When at least one element is NaN, the corresponding min value will be NaN. Args: a (cupy.ndarray): Array to take the maximum. axis (int): Along which axis to take the maximum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The maximum of ``a``, along the axis if specified. .. note:: When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements. .. seealso:: :func:`numpy.amax` z4cupy.amax does not support `keepdims` in fusion yet.rr )rr rrramaxmaxrrrrr3rrcC0tj||||d}t|rtdt|S)aReturns the minimum of an array along an axis ignoring NaN. When there is a slice whose elements are all NaN, a :class:`RuntimeWarning` is raised and NaN is returned. Args: a (cupy.ndarray): Array to take the minimum. axis (int): Along which axis to take the minimum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The minimum of ``a``, along the axis if specified. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.nanmin` r All-NaN slice encountered)rnanminrisnananywarningswarnRuntimeWarningrr r r resrrrrX rcCr)aReturns the maximum of an array along an axis ignoring NaN. When there is a slice whose elements are all NaN, a :class:`RuntimeWarning` is raised and NaN is returned. Args: a (cupy.ndarray): Array to take the maximum. axis (int): Along which axis to take the maximum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is remained as an axis of size one. Returns: cupy.ndarray: The maximum of ``a``, along the axis if specified. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.nanmax` r r)rnanmaxrrrrr r!r"rrrr%wr$r%cCs|j|||dS)aPReturns the range of values (maximum - minimum) along an axis. .. note:: The name of the function comes from the acronym for 'peak to peak'. When at least one element is NaN, the corresponding ptp value will be NaN. Args: a (cupy.ndarray): Array over which to take the range. axis (int): Axis along which to take the minimum. The flattened array is used by default. out (cupy.ndarray): Output array. keepdims (bool): If ``True``, the axis is retained as an axis of size one. Returns: cupy.ndarray: The minimum of ``a``, along the axis if specified. .. note:: When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements. .. seealso:: :func:`numpy.amin` r )ptprrrrr&sr&linearcst|}t|}|jdkr|d}d}nd}|jdkr&td|jt|tr.|f}|rO|dur:dj} ntj } |D] } d| | j<qAt | } |dura|rZ } n } d} nHt fdd|D}t tjt |} t| } tt| D] \}}||q|rj d| d } nj d| d } d }| j|d | j |}||d }|d vrtd|d|dkrt|tj}n;|dkrt|tj}n-|dkrdt|t|}n|dkrt|tj}n |dkr ntd||jtjkr't| |}|j|d|d}n=|dursz&_quantile_unchecked..)r.)r ?) inverted_cdfaveraged_inverted_cdfclosest_observationinterpolated_inverted_cdfhazenweibullmedian_unbiasednormal_unbiased'z;' method is not yet supported. Please use any other method.lowerhighermidpointg?nearestr'z^Unexpected interpolation method. Actual: '{}' not in ('linear', 'lower', 'higher', 'midpoint')r)dtypez0S idx, raw T a, raw int32 offset, raw int32 sizezU reta# ptrdiff_t idx_below = floor(idx); U weight_above = idx - idx_below; ptrdiff_t max_idx = size - 1; ptrdiff_t offset_bottom = _ind.get()[0] * offset + idx_below; ptrdiff_t offset_top = min(offset_bottom + 1, max_idx); U diff = a[offset_top] - a[offset_bottom]; if (weight_above < 0.5) { ret = a[offset_bottom] + diff * weight_above; } else { ret = a[offset_top] - diff * (1 - weight_above); } cupy_percentile_weightnening)$cupy result_typeasarrayr) ValueErrorformat isinstanceintlistshapetupleravelflattensetrangelen enumeratesortedswapaxesreshapecopysortfloorastypeint32ceilaroundr=rollaxistakeemptyElementwiseKernelsizesqueezer_internal_ascontiguousarray)rqr r overwrite_inputmethodr r=zerodkeepdimr+apnkeepkeepisNxindicesretrr,r_quantile_uncheckeds                "     rmcCs0t|tjrtnt}|d|ko||dkS)Ngr/)rDr?ndarraynumpy count_nonzero)r`xprrr_quantile_is_valid#srr) interpolationc CsZ|dur t||d}t|ttfrt|}|d}t|s"tdt|||||||dS)a Computes the q-th percentile of the data along the specified axis. Args: a (cupy.ndarray): Array for which to compute percentiles. q (float, tuple of floats or cupy.ndarray): Percentiles to compute in the range between 0 and 100 inclusive. axis (int or tuple of ints): Along which axis or axes to compute the percentiles. The flattened array is used by default. out (cupy.ndarray): Output array. overwrite_input (bool): If True, then allow the input array `a` to be modified by the intermediate calculations, to save memory. In this case, the contents of the input `a` after this function completes is undefined. method (str): Interpolation method when a quantile lies between two data points. ``linear`` interpolation is used by default. Supported interpolations are ``lower``, ``higher``, ``midpoint``, ``nearest`` and ``linear``. keepdims (bool): If ``True``, the axis is remained as an axis of size one. interpolation (str): Deprecated name for the method keyword argument. Returns: cupy.ndarray: The percentiles of ``a``, along the axis if specified. .. seealso:: :func:`numpy.percentile` N percentiledz)Percentiles must be in the range [0, 100]r r rarbr _check_interpolation_as_methodrDrHrFrorArrrBrmrr`r r rarbr rsrrrrt(s  rtc CsR|dur t||d}t|ttfrt|}t|stdt|||||||dS)aComputes the q-th quantile of the data along the specified axis. Args: a (cupy.ndarray): Array for which to compute quantiles. q (float, tuple of floats or cupy.ndarray): Quantiles to compute in the range between 0 and 1 inclusive. axis (int or tuple of ints): Along which axis or axes to compute the quantiles. The flattened array is used by default. out (cupy.ndarray): Output array. overwrite_input (bool): If True, then allow the input array `a` to be modified by the intermediate calculations, to save memory. In this case, the contents of the input `a` after this function completes is undefined. method (str): Interpolation method when a quantile lies between two data points. ``linear`` interpolation is used by default. Supported interpolations are ``lower``, ``higher``, ``midpoint``, ``nearest`` and ``linear``. keepdims (bool): If ``True``, the axis is remained as an axis of size one. interpolation (str): Deprecated name for the method keyword argument. Returns: cupy.ndarray: The quantiles of ``a``, along the axis if specified. .. seealso:: :func:`numpy.quantile` Nquantilez%Quantiles must be in the range [0, 1]rvrwryrrrrzYs  rzcCs,tjd|dtdd|dkrtd|S)Nz!the `interpolation=` argument to z was renamed to `method=`, which has additional options. Users of the modes 'nearest', 'lower', 'higher', or 'midpoint' are encouraged to review the method they. (Deprecated NumPy 1.22)) stacklevelr'zjYou shall not pass both `method` and `interpolation`! (`interpolation` is Deprecated in favor of `method`))rr DeprecationWarning TypeError)rbrsfnamerrrrxs rx)NNF)NNFr'F) __future__rrror?r cupy._corerrr cupy._logicrrrrr%r&rmrrrtrzrxrrrrsD       % %   n 1 1