o
    ÅÙ¾iT  ã                   @   s’   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZ dd	lmZmZ g d
¢Zdefdd„Zddedefdd„ZdS )é   )ÚLayout)ÚBlackwellMXScaleLayout)ÚBlackwellMXValueLayout)ÚHopperMXScaleLayout)ÚHopperMXValueLayout)ÚCDNA4MXScaleLayout)ÚStridedLayouté   )Úcuda_capability_geqÚis_hip_cdna4)r   r   r   r   r   r   r   Úmx_axisc                 C   s0   t dƒr	ttƒ fS t dƒrtd| ifS ttƒ fS )Né
   é	   r   )r
   r   Údictr   r   )r   © r   úX/home/ubuntu/.local/lib/python3.10/site-packages/triton_kernels/tensor_details/layout.pyÚ"make_default_matmul_mxfp4_w_layout   s
   

r   é   Ú	num_warpsc                 C   sB   t ƒ rttƒ fS tdƒrttƒ fS tdƒrt| |dœfS ttƒ fS )Nr   r   ©r   r   )r   r   r   r
   r   r   r   r   r   r   r   Ú(make_default_matmul_mxfp4_w_scale_layout   s   


r   N)r   )Úlayout_details.baser   Úlayout_details.blackwell_scaler   Úlayout_details.blackwell_valuer   Úlayout_details.hopper_scaler   Úlayout_details.hopper_valuer   Úlayout_details.cdna4_scaler   Úlayout_details.stridedr   Útarget_infor
   r   Ú__all__Úintr   r   r   r   r   r   Ú<module>   s    
