o XÛ·iíã@s<ddlmZddlZddlmZe dddd¡Zdd „ZdS) é)Ú annotationsN)Ú_utilz T A, N r, N czN outzout = A != 0 ? r - c : 0Úcupyx_scipy_linalg_band_posc Cs¾t |¡}|jdkr dSt |¡|jdr|j}n|}|j\}}tjd|…d|…f\}}t |||ƒ}|jdrLt t  |¡ƒ}t t  |¡ƒ }||fSt t  |¡ƒ}t t  |¡ƒ }||fS)aøReturn the lower and upper bandwidth of a 2D numeric array. Parameters ---------- a : ndarray Input array of size (M, N) Returns ------- lu : tuple 2-tuple of ints indicating the lower and upper bandwidth. A zero denotes no sub- or super-diagonal on that side (triangular), and, say for M rows (M-1) means that side is full. Same example applies to the upper triangular part with (N-1). .. seealso:: :func:`scipy.linalg.bandwidth` r)rrÚ F_CONTIGUOUS) ÚcupyÚasarrayÚsizerÚ _assert_2dÚflagsÚTÚshapeÚmgridÚ_kernel_cupy_band_pos_cÚintÚamaxÚamin) ÚaÚAÚmÚnÚrow_numÚcol_numÚbandptsÚ upper_bandÚ lower_band©rúU/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/linalg/_array_utils.pyÚ bandwidths"       ýr)Ú __future__rrÚ cupy.linalgrÚElementwiseKernelrrrrrrÚs  ü