o W۷i @sjddlmZddlmZddlmZddlZddZeddZed d Zed d Z dd dZ dddZ dS)) annotations)_core) bf16_loopNcCs8tjd|dgtddddddRd ||d S) Ncupy_ze->??zf->?zd->?zF->?zD->?zout0 = %s(in0))doc)r create_ufuncr)namerr I/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_logic/content.py_create_float_test_ufunc s  r isfinitezTests finiteness elementwise. Each element of returned array is ``True`` only if the corresponding element of the input is finite (i.e. not an infinity nor NaN). .. seealso:: :data:`numpy.isfinite` isinfzgTests if each element is the positive or negative infinity. .. seealso:: :data:`numpy.isinf` isnanzKTests if each element is a NaN. .. seealso:: :data:`numpy.isnan` c CsXt|}zt|}Wnty#}z |j}td|d|d}~wwtj|||dS)aTest element-wise for negative infinity, return result as bool array. Parameters ---------- x : cupy.ndarray Input array. out : cupy.ndarray, optional A location into which the result is stored. If provided, it should have a shape that input broadcasts to. By default, None, a freshly- allocated boolean array, is returned. Returns ------- y : cupy.ndarray Boolean array of same shape as ``x``. Examples -------- >>> cupy.isneginf(0) array(False) >>> cupy.isneginf(-cupy.inf) array(True) >>> cupy.isneginf(cupy.array([-cupy.inf, -4, cupy.nan, 0, 4, cupy.inf])) array([ True, False, False, False, False, False]) See Also -------- numpy.isneginf $This operation is not supported for & values because it would be ambiguous.Noutrcupysignbit TypeErrordtype logical_andxris_infrerr r r isneginf/s!rc CsZt|}zt|}Wnty$}z |j}td|d|d}~wwtj|||dS)aTest element-wise for positive infinity, return result as bool array. Parameters ---------- x : cupy.ndarray Input array. out : cupy.ndarray A location into which the result is stored. If provided, it should have a shape that input broadcasts to. By default, None, a freshly- allocated boolean array, is returned. Returns ------- y : cupy.ndarray Boolean array of same shape as ``x``. Examples -------- >>> cupy.isposinf(0) array(False) >>> cupy.isposinf(cupy.inf) array(True) >>> cupy.isposinf(cupy.array([-cupy.inf, -4, cupy.nan, 0, 4, cupy.inf])) array([False, False, False, False, False, True]) See Also -------- numpy.isposinf rrNrrrr r r isposinf\s!r )N) __future__rrr cupy._utilrr rrrrr r r r r s&     -