B \@s:ddlmZddlmZddlmZddlZdddZdS))cuda)driver) numpy_supportNc st|dd}|sJ|j\}}|jj||jjf}tjjj||f||j|d}t |jt j }t tdt|dd}t ||}||dftjfdd} t |jd|dt |jd|df} ||f} | | | |f|||S)aCompute the transpose of 'a' and store it into 'b', if given, and return it. If 'b' is not given, allocate a new array and return that. This implements the algorithm documented in http://devblogs.nvidia.com/parallelforall/efficient-matrix-transpose-cuda-cc/ :param a: an `np.ndarray` or a `DeviceNDArrayBase` subclass. If already on the device its stream will be used to perform the transpose (and to copy `b` to the device if necessary). streamr)dtyperc stjjd}tjj}tjj}tjjtjj}tjjtjj}||}||}|||jdkr|||jdkr|||||f|||f<t ||jdkr||jdkr|||f|||f<dS)N)shaperrr) rZsharedZarrayZ threadIdxxyZblockIdxZblockDimr Z syncthreads) inputoutputZtileZtxZtyZbxZbyr r )dt tile_shape;lib/python3.7/site-packages/numba/cuda/kernels/transpose.pykernel(s$ztranspose..kernel)getattrr ritemsizerZcudadrvZ devicearrayZ DeviceNDArraynpsZ from_dtyperZ get_deviceZMAX_THREADS_PER_BLOCKintmathpowlogZjit) abrZcolsZrowsstridesZtpbZ tile_widthZ tile_heightrZblocksZthreadsr)rrr transposes&      ,r)N)ZnumbarZnumba.cuda.cudadrv.driverrrrrrrrrrs