�
.��g�
� � � d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dl m
Z
ed�� � G d� de � � � � Z G d � d
e� � � Z
dS ) � N)�ABCMeta�abstractmethod�abstractclassmethod)� dataclass)�UnionT)�frozenc �> � e Zd ZU eed< eeef ed< eed< dS )� GPUTarget�backend�arch� warp_sizeN)�__name__�
__module__� __qualname__�str�__annotations__r �int� � �h/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/triton/backends/compiler.pyr
r
s8 � � � � � � � �L�L�L�
��S��/�����N�N�N�N�Nr r
c �� � e Zd Zdeddfd�Zedefd�� � Zedefd�� � Z e
defd�� � Z e
d e de
fd
�� � Ze
d e d e
ddfd �� � Ze
d
� � � ZdS )� BaseBackend�target�returnNc �B � || _ | � |� � sJ �d S �N)r �supports_target)�selfr s r �__init__zBaseBackend.__init__ s) � � �� ��#�#�F�+�+�+�+�+�+�+r �binaryc � � t j � t j � t � � t j � � }t j � d| � � � � d�d� � t j � |ddd| � � g}|D ]�}|� d� � d }t j �
|� � r�t j � |� � rst j
|d gt j �
� � }|�Ot j d |� d � � t j �
� � }|�||� d� � fc S ��t) d| � �� � �)N�TRITON_�_PATH� � third_party�cuda�bin� r z --version)�stderrz.*release (\d+\.\d+).*zutf-8)�flags� z Cannot find )�os�path�join�dirname�__file__�pardir�environ�get�upper�split�exists�isfile�
subprocess� check_output�STDOUT�re�search�decode� MULTILINE�group� RuntimeError)r �base_dir�paths�pr' �result�versions r �_path_to_binaryzBaseBackend._path_to_binary sC � ��7�<�<������ 9� 9�2�9�E�E�� �J�N�N�:�V�\�\�^�^�:�:�:�B� ?� ?� �G�L�L��=�&�%�� H� H�
�� � 3� 3�A��'�'�#�,�,�q�/�C��w�~�~�c�"�"�
3�r�w�~�~�c�':�':�
3�#�0�#�{�1C�J�L]�^�^�^���%� �i�(A�6�=�=�QX�CY�CY�ac�am� n� n� n�G� �*� �'�-�-��"2�"2� 2�2�2�2���2�&�2�2�3�3�3r c � � t �r ��NotImplementedError)r s r r zBaseBackend.supports_target* s � �!�!r c � � t �)z,Returns a unique identifier for this backendrH )r s r �hashzBaseBackend.hash. s
� � "�!r �optionsc � � t �)z�
Converts an `options` dictionary into an arbitrary object and returns it.
This function may contain target-specific heuristics and check the legality of the provided options
rH )r rL s r �
parse_optionszBaseBackend.parse_options3 s
� � "�!r �stagesc � � t �)a�
Populates `stages` dictionary with entries of the form:
ir_name [str] => Function[(src: str, metadata: dict) -> str|bytes]
The value of each entry may populate a `metadata` dictionary.
Stages will be run sequentially (in inseriton order) and can communicate using `metadata`.
All stages are expected to return a `str` object, except for the last stage which returns
a `bytes` object for execution by the launcher.
rH )r rO rL s r �
add_stageszBaseBackend.add_stages; s
� � "�!r c � � t �)zK
Load additional MLIR dialects into the provided `context`
rH )r �contexts r �
load_dialectszBaseBackend.load_dialectsG s
� �
"�!r )r r r r
r � staticmethodr rF r r r rK �dict�objectrN rQ rT r r r r r s* � � � � � �,�y� ,�T� ,� ,� ,� ,� �4�� 4� 4� 4� �\�4� �"� � "� "� "� ��"� �"�c� "� "� "� �^�"� �"�T� "�f� "� "� "� �^�"� � "�� "�� "�4� "� "� "� �^� "� �"� "� �^�"� "� "r r )� metaclass)r, r; r8 �abcr r r � dataclassesr �typingr rW r
r r r r �<module>r\ s� �� � � � � � � � � � � � � <� <� <� <� <� <� <� <� <� <� !� !� !� !� !� !� � � � � � � ��$����� � � � �� � � ���9"� 9"� 9"� 9"� 9"�G� 9"� 9"� 9"� 9"� 9"� 9"r