
    h	                     J    S SK Jr  S SKJrJr  S SKJr  \ " S S5      5       rg)    )	dataclass)ListAny)validate_block_shapec                   x    \ rS rSr% \\S'   \\   \S'   \\   \S'   \\   \S'   S r\	S\S\\   4S j5       r
S	rg
)TensorDescriptor   baseshapestridesblock_shapec                    [        U R                  5      n[        U R                  5      U:X  d
   SU  35       e[        U R                  5      U:X  d
   SU  35       eUS:  d   S5       eUS::  d   S5       e[	        U R
                  5      nUR                   SUR                   3nUS;  a(  U R
                  R                  5       S-  S:X  d   S	5       e[        U R                  5        U R
                  R                  R                  nU R                  S S
  H  nXT-  S-  S:X  a  M   S5       e   U R                  S
   S:X  d   S5       eg )Nzrank mismatch: r   zrank must not be zero   zrank cannot be more than 5.)ztorch.FakeTensorztorch.FunctionalTensor   zbase must be 16-byte alignedzstrides must be 16-byte aligned   z!Last dimension must be contiguous)lenr   r   r   typer
   
__module____name__data_ptrr   dtypeitemsize)selfrankty	type_name
elem_bytesstrides         X/var/www/fran/franai/venv/lib/python3.13/site-packages/triton/tools/tensor_descriptor.py__post_init__TensorDescriptor.__post_init__   s>   4::4<< D(BOD6*BB(4##$,Fv.FF,ax000xqy666y$))_}}oQr{{m4	JJ99%%'",1Q3QQ1T--.YY__--
ll3B'F'2-2U4UU2 (||B1$I&II$    tensorc                 N    [        U U R                  U R                  5       U5      $ )N)r   r   r    )r%   r   s     r!   from_tensorTensorDescriptor.from_tensor   s%    LLMMO	
 	
r$    N)r   r   __qualname____firstlineno__r   __annotations__r   intr"   staticmethodr'   __static_attributes__r)   r$   r!   r   r      sQ    
I9#YcJ  
C 
d3i 
 
r$   r   N)dataclassesr   typingr   r   triton._utilsr   r   r)   r$   r!   <module>r3      s'    !  . 
 
 
r$   