o Td @s:ddlmZddlmZddlZddlmZdddZdS))cuda)driverN) numpy_supportc st|dd}|s%|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}||}||}|||jdkrH|||jdkrH|||||f|||f<t ||jdkrf||jdkrh|||f|||f<dSdSdS)N)shaperrr) rsharedarray threadIdxxyblockIdxblockDimr syncthreads) inputoutputtiletxtybxbyr rdt tile_shapeo/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/cuda/kernels/transpose.pykernel)s$ztranspose..kernel)getattrr ritemsizercudadrv devicearray DeviceNDArraynps from_dtyper get_deviceMAX_THREADS_PER_BLOCKintmathpowlogjit) abrcolsrowsstridestpb tile_width tile_heightrblocksthreadsrrr transposes*      ,r7)N) numbarnumba.cuda.cudadrv.driverrr)numba.nprr$r7rrrrs