o
    پiO                     @   s2   d dl mZ d dlmZ ddeej fddZdS )	    )OptionalN   returnc                    s^   dk r
t | | d u rd S t fddtt |D }tjj| || | j d dS )Nr   c                 3   s(    | ]}|krt j d ndV  qdS ))divisibilityr   N)cute	sym_int64).0ir   leading_dim G/home/ubuntu/.local/lib/python3.10/site-packages/quack/compile_utils.py	<genexpr>   s
    
z#make_fake_tensor.<locals>.<genexpr>   )strideassumed_align)lentupleranger   runtimemake_fake_tensorwidth)dtypeshaper   r   r   r   r   r   r      s   
r   )r   r   )typingr   cutlass.cuter   Tensorr   r   r   r   r   <module>   s   