o
    ½Ù¾iý  ã                   @   sd   d dl Z ddlmZ eƒ  e jjjZe jjjZe jjjZe jjjZe jjj	Z	e jjjZ
e jjjZdS )é    Né   )Ú_ensure_initialized)ÚtorchÚ r   ÚopsÚ	deep_gemmÚget_tma_aligned_sizeÚ&get_mk_alignment_for_contiguous_layoutÚget_mn_major_tma_aligned_tensorÚ,get_mn_major_tma_aligned_packed_ue8m0_tensorÚ6get_k_grouped_mn_major_tma_aligned_packed_ue8m0_tensorÚ%get_m_alignment_for_contiguous_layoutÚ%get_k_alignment_for_contiguous_layout© r   r   úJ/home/ubuntu/.local/lib/python3.10/site-packages/deep_gemm/utils/layout.pyÚ<module>   s    





