o W۷i'@s6ddlmZddlZddlmZddZdddZdS) ) annotationsN)_corec Cspzt|}Wn ty|f}Ynwt|}|dt|kr.t|tjr.tj|d|dStj|d|d}||jkrDd|j||}tddt |j |D}|j dkr^tj ||j d Sg}g}t |j |D]6\}} | dkr{||||qh|dkr|||| qh|d|||| ||qhtj ||j d } | j rt||| | |S) a Construct an array by repeating A the number of times given by reps. Args: A (cupy.ndarray): Array to transform. reps (int or tuple): The number of repeats. Returns: cupy.ndarray: Transformed array with repeats. .. seealso:: :func:`numpy.tile` T)copyndminF)rcss|] \}}||VqdSN).0strrO/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_manipulation/tiling.py #sztile..r)dtype)tuple TypeErrorlencount isinstancecupyndarrayarrayndimzipshapesizeemptyrappendrelementwise_copyreshape) Arepstupdc shape_outc_shape ret_shapedim_innrepretrrr tiles>              r*cCs |||S)a+Repeat arrays along an axis. Args: a (cupy.ndarray): Array to transform. repeats (int, list or tuple): The number of repeats. axis (int): The axis to repeat. Returns: cupy.ndarray: Transformed array with repeats. .. seealso:: :func:`numpy.repeat` )repeat)arepeatsaxisrrr r+:s r+r) __future__rrrr*r+rrrr s  3