§
    OŠtjA  ã                   ó\   — d dl mZ d dlmZmZ d dlmZmZ e G d„ d¦  «        ¦   «         ZdS )é    )Ú	dataclass)ÚListÚAny)Úvalidate_block_shapeÚcanonicalize_dtypec                   ó¬   — e Zd ZU eed<   ee         ed<   ee         ed<   ee         ed<   dZeed<   dZ	e
ed<   d	„ Zedd
edee         fd„¦   «         ZdS )ÚTensorDescriptorÚbaseÚshapeÚstridesÚblock_shapeÚzeroÚpaddingFÚround_f32_to_tf32c                 ó„  — t          | j        ¦  «        }t          | j        ¦  «        |k    sJ d| › �¦   «         ‚t          | j        ¦  «        |k    sJ d| › �¦   «         ‚|dk    s
J d¦   «         ‚|dk    s
J d¦   «         ‚t	          | j        ¦  «        }|j        dvr*| j                             ¦   «         dz  dk    s
J d¦   «         ‚t          | j        ¦  «         | j        j	        j
        }| j        d d	…         D ]}||z  dz  dk    s
J d
¦   «         ‚Œ| j        D ]}|dk    s
J d¦   «         ‚Œ| j        d	         dk    s
J d¦   «         ‚| j        dk    s| j        dk    s
J d¦   «         ‚| j        dk    r| j        j	        j        s
J d¦   «         ‚| j        r)t          | j        j	        ¦  «        }|dk    sJ d¦   «         ‚d S d S )Nzrank mismatch: r   zrank must not be zeroé   zrank cannot be more than 5)Ú
FakeTensorÚFunctionalTensoré   zbase must be 16-byte alignedéÿÿÿÿzstrides must be 16-byte alignedzshape must be positiveé   z!Last dimension must be contiguousr   ÚnanzIllegal value for paddingzAPadding option `nan` is only supported for floating point tensorsÚfp32z7round_f32_to_tf32 is only supported for float32 tensors)Úlenr   r   r   Útyper
   Ú__name__Údata_ptrr   ÚdtypeÚitemsizer   Úis_floating_pointr   r   )ÚselfÚrankÚtyÚ
elem_bytesÚstrideÚ	shape_dimÚ
dtype_names          ú\/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/triton/tools/tensor_descriptor.pyÚ__post_init__zTensorDescriptor.__post_init__   s  € Ý�4”:‰ŒˆÝ�4”<Ñ Ô  DÒ(Ð(Ð(Ð*B¸DÐ*BÐ*BÑ(Ô(Ð(Ý�4Ô#Ñ$Ô$¨Ò,Ð,Ð,Ð.FÀÐ.FÐ.FÑ,Ô,Ð,Ø�aŠxˆxˆxÐ0‰xŒxˆxØ�qŠyˆyˆyÐ6‰yŒyˆyÝ�$”)‰_Œ_ˆØŒ;Ð@Ð@Ð@Ø”9×%Ò%Ñ'Ô'¨"Ñ,°Ò1Ð1Ð1Ð3QÑ1Ô1Ð1Ý˜TÔ-Ñ.Ô.Ð.Ø”Y”_Ô-ˆ
Ø”l 3 B 3Ô'ð 	Vð 	VˆFØ˜ZÑ'¨2Ñ-°Ò2Ð2Ð2Ð4UÑ2Ô2Ð2Ð2Øœð 	;ð 	;ˆIØ˜q’=�=�=Ð":‘=”=�=�=ØŒ|˜BÔ 1Ò$Ð$Ð$Ð&IÑ$Ô$Ð$ØŒ|˜vÒ%Ð%¨¬¸Ò)>Ð)>Ð)>Ð@[Ñ)>Ô)>Ð>ØŒ<˜5Ò Ð Ø”9”?Ô4ÐyÐyÐ6yÑyÔyÐ4ØÔ!ð 	cÝ+¨D¬I¬OÑ<Ô<ˆJØ Ò'Ð'Ð'Ð)bÑ'Ô'Ð'ð	cð 	cà'Ð'ó    Útensorc                 óX   — t          | | j        |                      ¦   «         |||¦  «        S )N)r	   r   r%   )r+   r   r   r   s       r(   Úfrom_tensorzTensorDescriptor.from_tensor&   s&   € å ¨¬°f·m²m±o´oÀ{ÐT[Ð]nÑoÔoÐor*   N)r   F)r   Ú
__module__Ú__qualname__r   Ú__annotations__r   Úintr   Ústrr   Úboolr)   Ústaticmethodr-   © r*   r(   r	   r	      sÂ   € € € € € € à
€I€I�IØ�Œ9ÐÐÑØ�#ŒYÐÐÑØ�c”ÐÐÑØ€GˆSÐÐÑØ#Ð�tÐ#Ð#Ñ#ðcð cð cð. ðpð p˜Cð p¨d°3¬ið pð pð pñ „\ðpð pð pr*   r	   N)	Údataclassesr   Útypingr   r   Útriton._utilsr   r   r	   r5   r*   r(   ú<module>r9      s�   ðØ !Ð !Ð !Ð !Ð !Ð !Ø Ð Ð Ð Ð Ð Ð Ð Ø BÐ BÐ BÐ BÐ BÐ BÐ BÐ Bð ð!pð !pð !pð !pð !pñ !pô !pñ „ð!pð !pð !pr*   