o
    ÒÙ¾iž
  ã                	   @   sh   d Z ddlmZ ejZejZejZej	Z
ejZejZejZejZejZeeee
eeeeeg	Zdd„ ZdS )z!
Registry of swizzling functions
é    )ÚSwizzlingFunctorc                   C   s   t S )N)Ú_swizzling_functors© r   r   úi/home/ubuntu/.local/lib/python3.10/site-packages/flashinfer/data/cutlass/python/cutlass_cppgen/swizzle.pyÚget_swizzling_functors@   s   r   N)Ú__doc__Úcutlass_libraryr   Ú	Identity1ÚIdentitySwizzle1Ú	Identity2ÚIdentitySwizzle2Ú	Identity4ÚIdentitySwizzle4Ú	Identity8ÚIdentitySwizzle8Ú
HorizontalÚHorizontalSwizzleÚStreamKÚThreadblockSwizzleStreamKÚStridedDgradIdentity1ÚStridedDgradIdentitySwizzle1ÚStridedDgradIdentity4ÚStridedDgradIdentitySwizzle4ÚStridedDgradHorizontalÚStridedDgradHorizontalSwizzler   r   r   r   r   r   Ú<module>   s,    ÷