§
    ÐóÉj§L  ã            
      ó6  — U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	m
Z
 d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZ er"ddlmZmZ dd	l m!Z!m"Z"m#Z# dd
l$m%Z% ddl&m'Z'm(Z(m)Z)  G d„ d¦  «        Z* e*¦   «         Z+dZ,de-d<   dcd„Z.ddd„Z/ ed¦  «        Z0 ed¦  «        Z1 e2¦   «         Z3 G d„ dee0e1f         ¦  «        Z4 G d „ d!e4e5e5f         ¦  «        Z6 G d"„ d#e4e5e5f         ¦  «        Z7 G d$„ d%e4e8e8f         ¦  «        Z9 G d&„ d'e4e:e:f         ¦  «        Z; ed(¦  «        Z< G d)„ d*ee<         e4eee<                  eee<                  f         ¦  «        Z=e
 G d+„ d,¦  «        ¦   «         Z> G d-„ d.e4e5e>f         ¦  «        Z? G d/„ d0e4ee5         ee5         f         ¦  «        Z@ G d1„ d2e4¦  «        ZA e
d¬3¦  «         G d4„ d5¦  «        ¦   «         ZB G d6„ d7e¦  «        ZC ed8d9¬:¦  «        ZD G d;„ d9¦  «        ZE G d<„ d=e¦  «        ZF G d>„ d?eE¦  «        ZG G d@„ dAeE¦  «        ZHdBe-dC<    G dD„ dBeE¦  «        ZI G dE„ dFeE¦  «        ZJ G dG„ dHe¦  «        ZK G dI„ dJeE¦  «        ZL G dK„ dLe¦  «        ZM G dM„ dNe¦  «        ZN edOe¬:¦  «        ZO G dP„ dQeeO         ¦  «        ZPeQee5e:f                  ZR G dR„ dSe¦  «        ZS G dT„ dUe¦  «        ZT G dV„ dWe¦  «        ZU G dX„ dYeE¦  «        ZV G dZ„ d[eE¦  «        ZW G d\„ d]eE¦  «        ZX G d^„ d_eE¦  «        ZY G d`„ daeE¦  «        ZZ eG¦   «         Z[ eH¦   «         Z\ eI¦   «         Z] eJ¦   «         Z^ eL¦   «         Z_ eV¦   «         Z` eW¦   «         Za eX¦   «         Zb eY¦   «         Zc eZ¦   «         Zddb„ ZedS )eé    )ÚannotationsN)Ú	dataclass)Úcontextmanager)ÚcastÚAnyÚCallableÚ	GeneratorÚGenericÚOptionalÚProtocolÚTypeÚTypeVarÚ	TypedDictÚTYPE_CHECKINGÚUnion)ÚgetenvÚgetenv_boolé   )ÚCacheManagerÚRemoteCacheBackend)ÚJitFunctionInfoÚKernelParamÚJITFunction)ÚConfig)Ú	ASTSourceÚLazyDictÚIRSourcec                  ó   — e Zd ZdS )ÚEnvN)Ú__name__Ú
__module__Ú__qualname__© ó    úq/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/triton/knobs.pyr   r      s   € € € € € Ø€Dr$   r   TÚboolÚpropagate_envÚkeyÚstrÚvalueúOptional[str]ÚreturnÚNonec                óx   — t           sd S |�|t          j        | <   d S | t          j        v rt          j        | = d S d S ©N)r'   ÚosÚenviron)r(   r*   s     r%   Úsetenvr2   !   sF   € Ýð ØˆàÐØ�Œ
�3‰ˆˆØ	•”
Ð	Ð	ÝŒJ�sˆOˆOˆOð 
Ð	r$   Úvalr   ú!Union[None, tuple[Optional[str]]]c                ó–   — | €dS t          | ¦  «        }|t          u r| rdndfS |t          u r| fS |t          u rt          | ¦  «        fS d S )Nr/   Ú1Ú0)Útyper&   r)   Úint)r3   Úts     r%   Útoenvr;   +   s^   € Ø
€{ØˆxåˆS‰	Œ	€AØ�D€y€yØÐ#�� Ð&Ð&à�C€x€xØˆwˆà�C€x€xÝ�C‘”ˆ|Ðàˆ4r$   ÚSetTypeÚGetTypec                  óF   — e Zd Zdd„Zdd	„Zdd„Zdd„Zdd„Zdd„Zdd„Z	dS ) Úenv_baser(   r)   r,   r-   c                ó   — || _         d S r/   )r(   )Úselfr(   s     r%   Ú__init__zenv_base.__init__F   s   € ØˆŒˆˆr$   ÚobjclassúType[object]Únamec                ó   — || _         d S r/   )rE   )rA   rC   rE   s      r%   Ú__set_name__zenv_base.__set_name__I   s   € ØˆŒ	ˆ	ˆ	r$   ÚobjúOptional[object]úOptional[Type[object]]r=   c                ó°   — |j                              | j        t          ¦  «        }|t          u r|                      ¦   «         S |                      |¦  «        S r/   )Ú__dict__ÚgetrE   Ú_NOTHINGÚ	transform)rA   rH   rC   Úpy_vals       r%   Ú__get__zenv_base.__get__L   sF   € Ø”×!Ò! $¤)­XÑ6Ô6ˆØ•XÐÐØ—8’8‘:”:ÐØ�~Š~˜fÑ%Ô%Ð%r$   c                ó   — t          ¦   «         ‚r/   )ÚNotImplementedError©rA   s    r%   rM   zenv_base.getR   s   € Ý!Ñ#Ô#Ð#r$   Úobjectr*   úUnion[SetType, Env]c                óî   — t          |t          ¦  «        r"|j                             | j        d ¦  «         d S ||j        | j        <   t          |¦  «        x}rt          | j        |d         ¦  «         d S d S )Nr   )Ú
isinstancer   rL   ÚpoprE   r;   r2   r(   )rA   rH   r*   Úenv_vals       r%   Ú__set__zenv_base.__set__U   s{   € Ý�e�SÑ!Ô!ð 	-ØŒL×Ò˜TœY¨Ñ-Ô-Ð-Ð-Ð-à&+ˆCŒL˜œÑ#Ý ™,œ,Ð&ˆwð -Ý�t”x ¨¤Ñ,Ô,Ð,Ð,Ð,ð-ð -r$   c                óF   — |j                              | j        d ¦  «         d S r/   )rL   rY   rE   )rA   rH   s     r%   Ú
__delete__zenv_base.__delete__]   s"   € ØŒ×Ò˜œ DÑ)Ô)Ð)Ð)Ð)r$   r3   r<   c                ó,   — t          t          |¦  «        S r/   )r   r=   )rA   r3   s     r%   rO   zenv_base.transform`   s   € õ •G˜SÑ!Ô!Ð!r$   N)r(   r)   r,   r-   )rC   rD   rE   r)   r,   r-   )rH   rI   rC   rJ   r,   r=   )r,   r=   )rH   rU   r*   rV   r,   r-   )rH   rU   r,   r-   )r3   r<   r,   r=   )
r    r!   r"   rB   rG   rQ   rM   r[   r]   rO   r#   r$   r%   r?   r?   D   s    € € € € € ðð ð ð ðð ð ð ð&ð &ð &ð &ð$ð $ð $ð $ð-ð -ð -ð -ð*ð *ð *ð *ð"ð "ð "ð "ð "ð "r$   r?   c                  ó(   ‡ — e Zd Zdˆ fd„Zdd„Zˆ xZS )	Úenv_strr(   r)   Údefaultc                óX   •— t          ¦   «                              |¦  «         || _        d S r/   ©ÚsuperrB   ra   ©rA   r(   ra   Ú	__class__s      €r%   rB   zenv_str.__init__h   ó&   ø€ Ý‰Œ×Ò˜ÑÔÐØˆŒˆˆr$   r,   c                ó6   — t          | j        | j        ¦  «        S r/   )r   r(   ra   rT   s    r%   rM   zenv_str.getl   s   € Ý�d”h ¤Ñ-Ô-Ð-r$   )r(   r)   ra   r)   ©r,   r)   ©r    r!   r"   rB   rM   Ú__classcell__©rf   s   @r%   r`   r`   f   sQ   ø€ € € € € ðð ð ð ð ð ð.ð .ð .ð .ð .ð .ð .ð .r$   r`   c                  ó(   ‡ — e Zd Zdˆ fd„Zd	d„Zˆ xZS )
Úenv_str_callable_defaultr(   r)   Údefault_factoryúCallable[[], str]c                óX   •— t          ¦   «                              |¦  «         || _        d S r/   )rd   rB   ro   )rA   r(   ro   rf   s      €r%   rB   z!env_str_callable_default.__init__r   s)   ø€ Ý‰Œ×Ò˜ÑÔÐØ.ˆÔÐÐr$   r,   c                óZ   — t          | j        ¦  «        }|€|                      ¦   «         S |S r/   )r   r(   ro   )rA   rZ   s     r%   rM   zenv_str_callable_default.getv   s-   € Ý˜œÑ"Ô"ˆØˆ?Ø×'Ò'Ñ)Ô)Ð)Øˆr$   )r(   r)   ro   rp   ri   rj   rl   s   @r%   rn   rn   p   sQ   ø€ € € € € ð/ð /ð /ð /ð /ð /ðð ð ð ð ð ð ð r$   rn   c                  ó*   ‡ — e Zd Zd
dˆ fd„Zdd	„Zˆ xZS )Úenv_boolFr(   r)   ra   r&   r,   r-   c                óX   •— t          ¦   «                              |¦  «         || _        d S r/   rc   re   s      €r%   rB   zenv_bool.__init__   rg   r$   c                ó6   — t          | j        | j        ¦  «        S r/   )r   r(   ra   rT   s    r%   rM   zenv_bool.getƒ   s   € Ý˜4œ8 T¤\Ñ2Ô2Ð2r$   ©F)r(   r)   ra   r&   r,   r-   )r,   r&   rj   rl   s   @r%   rt   rt   }   sV   ø€ € € € € ðð ð ð ð ð ð ð3ð 3ð 3ð 3ð 3ð 3ð 3ð 3r$   rt   c                  ó*   ‡ — e Zd Zd
dˆ fd„Zdd	„Zˆ xZS )Úenv_intr   r(   r)   ra   r9   r,   r-   c                óX   •— t          ¦   «                              |¦  «         || _        d S r/   rc   re   s      €r%   rB   zenv_int.__init__‰   rg   r$   c                ó¸   — t          | j        ¦  «        }|€| j        S 	 t          |¦  «        S # t          $ r!}t          d| j        › d|› d�¦  «        |‚d }~ww xY w)NzUnable to use ú=z: expected int)r   r(   ra   r9   Ú
ValueErrorÚRuntimeError)rA   r3   Úexcs      r%   rM   zenv_int.get�   sy   € Ý�T”XÑÔˆØˆ;Ø”<Ðð	YÝ�s‘8”8ˆOøÝð 	Yð 	Yð 	YÝÐN°´ÐNÐN¸3ÐNÐNÐNÑOÔOÐUXÐXøøøøð	Yøøøs   Ÿ. ®
A¸AÁA)r   )r(   r)   ra   r9   r,   r-   ©r,   r9   rj   rl   s   @r%   ry   ry   ‡   s^   ø€ € € € € ðð ð ð ð ð ð ðYð Yð Yð Yð Yð Yð Yð Yr$   ry   Ú	ClassTypec                  ó(   ‡ — e Zd Zd	ˆ fd„Zd
d„Zˆ xZS )Ú	env_classr(   r)   r8   r,   r-   c                óX   •— t          ¦   «                              |¦  «         || _        d S r/   )rd   rB   r8   )rA   r(   r8   rf   s      €r%   rB   zenv_class.__init__œ   s&   ø€ Ý‰Œ×Ò˜ÑÔÐàˆŒ	ˆ	ˆ	r$   úOptional[Type[ClassType]]c           	     óø  ‡ — t          ‰ j        ¦  «        }|€d S |                     dd¦  «        }t          |¦  «        dk    rt	          d‰ j        › d|› d�¦  «        ‚t          t          j        |d         ¦  «        |d         ¦  «        }t          ˆ fd„| 	                    ¦   «         D ¦   «         ¦  «        s#t	          d	|› d
‰ j        › d‰ j
        › d�¦  «        ‚t          t          t                   |¦  «        S )Nú:r   é   zUnable to read z: 'z ' isn't of the form MODULE:CLASSr   c              3  ó8   •K  — | ]}|j         ‰j        k    V — Œd S r/   )r    r8   )Ú.0ÚcrA   s     €r%   ú	<genexpr>z env_class.get.<locals>.<genexpr>ª   s,   øè è € Ð?Ð?°�A”J $¤)Ò+Ð?Ð?Ð?Ð?Ð?Ð?r$   zUnable to use 'z' from z: not of type 'ú')r   r(   ÚsplitÚlenr~   ÚgetattrÚ	importlibÚimport_moduleÚanyÚmror8   r   r   r�   )rA   r3   ÚcompsÚclss   `   r%   rM   zenv_class.get¡   sö   ø€ Ý�T”XÑÔˆØˆ;Ø�4Ø—	’	˜#˜qÑ!Ô!ˆÝˆu‰:Œ:˜Š?ˆ?ÝÐc°´ÐcÐc¸cÐcÐcÐcÑdÔdÐdÝ•iÔ-¨e°A¬hÑ7Ô7¸¸q¼ÑBÔBˆåÐ?Ð?Ð?Ð?°S·W²W±Y´YÐ?Ñ?Ô?Ñ@Ô@ð 	dÝÐb°ÐbÐb¸T¼XÐbÐbÐVZÔV_ÐbÐbÐbÑcÔcÐcå•D�”O SÑ)Ô)Ð)r$   )r(   r)   r8   r)   r,   r-   )r,   r…   rj   rl   s   @r%   rƒ   rƒ   š   sQ   ø€ € € € € ðð ð ð ð ð ð
*ð *ð *ð *ð *ð *ð *ð *r$   rƒ   c                  óV   — e Zd ZU ded<   ded<   eej        dd„¦   «         ¦   «         ZdS )	Ú
NvidiaToolr)   ÚpathÚversionr,   úOptional[NvidiaTool]c                ó8  — 	 t          j        | dgt           j        ¬¦  «        }t          j        d|                     d¦  «        t          j        ¬¦  «        }|€d S t          | |                     d¦  «        ¦  «        S # t           j	        t          f$ r Y d S w xY w)Nz	--version)Ústderrz.*release (\d+\.\d+).*zutf-8)Úflagsr   )Ú
subprocessÚcheck_outputÚSTDOUTÚreÚsearchÚdecodeÚ	MULTILINEr˜   ÚgroupÚCalledProcessErrorÚFileNotFoundError)r™   Úresultrš   s      r%   Ú	from_pathzNvidiaTool.from_pathµ   sœ   € ð	ÝÔ,¨d°KÐ-@ÍÔIZÐ[Ñ[Ô[ˆFÝ”iÐ 9¸6¿=º=ÈÑ;QÔ;QÕY[ÔYeÐfÑfÔfˆGØˆØ�tÝ˜d G§M¢M°!Ñ$4Ô$4Ñ5Ô5Ð5øÝÔ-Õ/@ÐAð 	ð 	ð 	Ø�4�4ð	øøøs   ‚AA? Á"A? Á?BÂBN)r™   r)   r,   r›   )r    r!   r"   Ú__annotations__ÚstaticmethodÚ	functoolsÚ	lru_cacherª   r#   r$   r%   r˜   r˜   °   sT   € € € € € € à€I€I�IØ€L€L�LàØÔðð ð ñ Ôñ „\ðð ð r$   r˜   c                  ó0   ‡ — e Zd Zd
ˆ fd„Zdd„Zdd	„Zˆ xZS )Úenv_nvidia_toolÚbinaryr)   r,   r-   c                ón  •— |t          j        d¦  «        z  }|| _        t          j                             t          j                             t          ¦  «        ddd|¦  «        | _        t          ¦   «          
                    d|                     ¦   «                              dd¦  «        › d�¦  «         d S )	NÚEXEÚbackendsÚnvidiaÚbinÚTRITON_ú-Ú_Ú_PATH)Ú	sysconfigÚget_config_varr±   r0   r™   ÚjoinÚdirnameÚ__file__Údefault_pathrd   rB   ÚupperÚreplace)rA   r±   rf   s     €r%   rB   zenv_nvidia_tool.__init__Ä   s�   ø€ Ø•)Ô*¨5Ñ1Ô1Ñ1ˆØˆŒÝœGŸLšL­¬¯ª½Ñ)BÔ)BÀJÐPXÐZ_ÐagÑhÔhˆÔå‰Œ×ÒÐJ 6§<¢<¡>¤>×#9Ò#9¸#¸sÑ#CÔ#CÐJÐJÐJÑKÔKÐKÐKÐKr$   r˜   c                óP   — |                       t          | j        ¦  «        ¦  «        S r/   )rO   r   r(   rT   s    r%   rM   zenv_nvidia_tool.getË   s   € Ø�~Š~�f T¤XÑ.Ô.Ñ/Ô/Ð/r$   r™   c                ó¢   — |�
|| j         g}n| j         g}|D ]"}t                               |¦  «        x}r|c S Œ#t          d| j        › �¦  «        ‚)NzCannot find )rÀ   r˜   rª   r~   r±   )rA   r™   ÚpathsÚtools       r%   rO   zenv_nvidia_tool.transformÎ   sw   € ð ÐØ˜4Ô,Ð-ˆEˆEàÔ&Ð'ˆEàð 	ð 	ˆDÝ!×+Ò+¨DÑ1Ô1Ð1ˆtð Ø���ðõ Ð7¨$¬+Ð7Ð7Ñ8Ô8Ð8r$   )r±   r)   r,   r-   )r,   r˜   )r™   r)   r,   r˜   )r    r!   r"   rB   rM   rO   rk   rl   s   @r%   r°   r°   Â   sk   ø€ € € € € ðLð Lð Lð Lð Lð Lð0ð 0ð 0ð 0ð9ð 9ð 9ð 9ð 9ð 9ð 9ð 9r$   r°   c                  ó   — e Zd Zdd„ZdS )Úenv_opt_strr,   r+   c                ó*   — t          | j        ¦  «        S r/   )r   r(   rT   s    r%   rM   zenv_opt_str.getà   s   € Ý�d”hÑÔÐr$   N©r,   r+   ©r    r!   r"   rM   r#   r$   r%   rÈ   rÈ   Þ   s(   € € € € € ð ð  ð  ð  ð  ð  r$   rÈ   c                  ó   — e Zd Zdd„ZdS )Úenv_opt_boolr,   r+   c                ó,   — t          | j        d ¦  «        S r/   )r   r(   rT   s    r%   rM   zenv_opt_bool.getæ   s   € Ý˜4œ8 TÑ*Ô*Ð*r$   NrÊ   rË   r#   r$   r%   rÍ   rÍ   ä   s(   € € € € € ð+ð +ð +ð +ð +ð +r$   rÍ   )Úfrozenc                  ób   — e Zd ZU dZded<   ded<   ded<   edd„¦   «         Zedd	„¦   «         Zd
S )ÚCompileTimeszm
    Model holding timing information for an invocation of the compiler.

    All times in microseconds.
    r9   Úir_initializationzlist[tuple[str, int]]Úlowering_stagesÚstore_resultsr,   c                ó>   — t          d„ | j        D ¦   «         ¦  «        S )Nc              3  ó&   K  — | ]}|d          V — ŒdS )r   Nr#   )rŠ   Ústages     r%   rŒ   z.CompileTimes.total_lowering.<locals>.<genexpr>þ   s&   è è € Ð?Ð? �E˜!”HÐ?Ð?Ð?Ð?Ð?Ð?r$   )ÚsumrÓ   rT   s    r%   Útotal_loweringzCompileTimes.total_loweringü   s"   € åÐ?Ð?¨$Ô*>Ð?Ñ?Ô?Ñ@Ô@Ð@r$   c                ó0   — | j         | j        z   | j        z   S r/   )rÒ   rÙ   rÔ   rT   s    r%   ÚtotalzCompileTimes.total   s   € àÔ%¨Ô(;Ñ;¸dÔ>PÑPÐPr$   Nr€   )r    r!   r"   Ú__doc__r«   ÚpropertyrÙ   rÛ   r#   r$   r%   rÑ   rÑ   ê   s’   € € € € € € ðð ð ÐÐÑð +Ð*Ð*Ñ*ð ÐÐÑàðAð Að Añ „XðAð ðQð Qð Qñ „XðQð Qð Qr$   rÑ   c                  ó   — e Zd Zdd„ZdS )ÚCompilationListenerÚsrcúUnion[ASTSource, IRSource]Úmetadataúdict[str, Any]Úmetadata_groupúdict[str, str]ÚtimesrÑ   Ú	cache_hitr&   r,   r-   c               ó   — d S r/   r#   )rA   rà   râ   rä   ræ   rç   s         r%   Ú__call__zCompilationListener.__call__  ó   € àˆr$   N)rà   rá   râ   rã   rä   rå   ræ   rÑ   rç   r&   r,   r-   ©r    r!   r"   ré   r#   r$   r%   rß   rß     ó(   € € € € € ðð ð ð ð ð r$   rß   Ú
knobs_typeÚ
base_knobs)Úboundc                  óf   — e Zd Zedd„¦   «         Zedd„¦   «         Zdd„Zdd	„Zedd„¦   «         Z	dS )rî   r,   údict[str, env_base]c                ób   — d„ t          | ¦  «        j                             ¦   «         D ¦   «         S )Nc                óD   — i | ]\  }}t          |t          ¦  «        ¯||“ŒS r#   )rX   r?   )rŠ   ÚkÚvs      r%   ú
<dictcomp>z/base_knobs.knob_descriptors.<locals>.<dictcomp>  s@   € ð 
ð 
ð 
ñ ��1Ý˜!�XÑ&Ô&ð	
Øˆqð
ð 
ð 
r$   )r8   rL   ÚitemsrT   s    r%   Úknob_descriptorszbase_knobs.knob_descriptors  s:   € ð
ð 
õ ˜T™
œ
Ô+×1Ò1Ñ3Ô3ð
ñ 
ô 
ð 	
r$   rã   c                óN   ‡ — ˆ fd„‰ j                              ¦   «         D ¦   «         S )Nc                ó2   •— i | ]}|t          ‰|¦  «        “ŒS r#   )r�   )rŠ   rô   rA   s     €r%   rö   z$base_knobs.knobs.<locals>.<dictcomp>  s%   ø€ ÐJÐJÐJ¨�•7˜4 Ñ#Ô#ÐJÐJÐJr$   )rø   ÚkeysrT   s   `r%   Úknobszbase_knobs.knobs  s,   ø€ àJÐJÐJÐJ¨TÔ-B×-GÒ-GÑ-IÔ-IÐJÑJÔJÐJr$   rA   rí   c                ór   —  t          | ¦  «        ¦   «         }|j                             | j        ¦  «         |S r/   )r8   rL   Úupdate)rA   Úress     r%   Úcopyzbase_knobs.copy  s0   € Ø�d�4‰jŒj‰lŒlˆØŒ×Ò˜DœMÑ*Ô*Ð*Øˆ
r$   c                ó^   — | j                              ¦   «         D ]}t          | |¦  «         Œ| S r/   )rø   rû   Údelattr)rA   Úknobs     r%   Úresetzbase_knobs.reset#  s8   € ØÔ)×.Ò.Ñ0Ô0ð 	 ð 	 ˆDÝ�D˜$ÑÔÐÐØˆr$   úGenerator[None, None, None]c              #  ót  K  — 	 d„ | j                              ¦   «         D ¦   «         }t          | j        ¦  «        }d V — | j                             ¦   «          | j                             |¦  «         |                     ¦   «         D ]2\  }}|�|t          j        |<   Œ|t          j        v rt          j        |= Œ3d S # | j                             ¦   «          | j                             |¦  «         |                     ¦   «         D ]2\  }}|�|t          j        |<   Œ|t          j        v rt          j        |= Œ3w xY w)Nc                óB   — i | ]}|j         t          |j         ¦  «        “ŒS r#   )r(   r   )rŠ   r  s     r%   rö   z$base_knobs.scope.<locals>.<dictcomp>+  s&   € ÐaÐaÐa¸$˜4œ8¥V¨D¬HÑ%5Ô%5ÐaÐaÐar$   )	rø   ÚvaluesÚdictrL   Úclearrþ   r÷   r0   r1   )rA   Úinitial_envÚorigrô   rõ   s        r%   Úscopezbase_knobs.scope(  s:  è è € ð	&ØaÐaÀ$ÔBW×B^ÒB^ÑB`ÔB`ÐaÑaÔaˆKÝ˜œÑ&Ô&ˆDØˆEˆEˆEàŒM×ÒÑ!Ô!Ð!ØŒM× Ò  Ñ&Ô&Ð&à#×)Ò)Ñ+Ô+ð &ð &‘��1Ø�=Ø$%•B”J˜q‘M�MØ�"œ*�_�_Ýœ
 1˜øð	&ð &øð ŒM×ÒÑ!Ô!Ð!ØŒM× Ò  Ñ&Ô&Ð&à#×)Ò)Ñ+Ô+ð &ð &‘��1Ø�=Ø$%•B”J˜q‘M�MØ�"œ*�_�_Ýœ
 1˜øð	&øøøs   „;B; Â;A<D7N)r,   rñ   )r,   rã   )rA   rí   r,   rí   )r,   r  )
r    r!   r"   rÝ   rø   rü   r   r  r   r  r#   r$   r%   rî   rî     sž   € € € € € àð
ð 
ð 
ñ „Xð
ð ðKð Kð Kñ „XðKðð ð ð ð
ð ð ð ð
 ð&ð &ð &ñ „^ð&ð &ð &r$   c                  ó   — e Zd Zdd
„ZdS )Ú	BuildImplrE   r)   rà   ÚsrcdirÚlibrary_dirsú	list[str]Úinclude_dirsÚ	librariesr,   c               ó   — d S r/   r#   )rA   rE   rà   r  r  r  r  s          r%   ré   zBuildImpl.__call__;  rê   r$   N)rE   r)   rà   r)   r  r)   r  r  r  r  r  r  r,   r)   rë   r#   r$   r%   r  r  9  rì   r$   r  c                  óš   — e Zd ZU dZ ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<   d	Z	d
ed<   e
dd„¦   «         Zd	S )Úbuild_knobsz<Configuration controlling how the native compiler is invokedÚCCrÈ   ÚccÚTRITON_CUDACRT_PATHÚcudacrt_pathÚTRITON_CUDART_PATHÚcudart_pathNzOptional[BuildImpl]Úimplr,   úset[str]c                ó2   — d„ | j         | j        fD ¦   «         S )Nc                ó   — h | ]}|®|’ŒS r/   r#   )rŠ   r™   s     r%   ú	<setcomp>z+build_knobs.backend_dirs.<locals>.<setcomp>K  s   € Ð[Ð[Ð[˜È$ÐJZ�ÐJZÐJZÐJZr$   )r  r  rT   s    r%   Úbackend_dirszbuild_knobs.backend_dirsI  s!   € à[Ð[ $Ô"3°TÔ5EÐ!FÐ[Ñ[Ô[Ð[r$   )r,   r  )r    r!   r"   rÜ   rÈ   r  r«   r  r  r  rÝ   r#  r#   r$   r%   r  r  @  s¤   € € € € € € ØFÐFØ!�k $Ñ'Ô'€BÐ'Ð'Ð'Ñ'à + Ð,AÑ BÔ B€LÐBÐBÐBÑBØ*˜{Ð+?Ñ@Ô@€KÐ@Ð@Ð@Ñ@à $€DÐ$Ð$Ð$Ñ$àð\ð \ð \ñ „Xð\ð \ð \r$   r  c                  óv   — e Zd ZU  edd¦  «        Zded<    edd¦  «        Zded<    edd	¦  «        Zd
ed<   dS )Úredis_knobsÚTRITON_REDIS_KEY_FORMATztriton:{key}:{filename}r`   Ú
key_formatÚTRITON_REDIS_HOSTÚ	localhostÚhostÚTRITON_REDIS_PORTië  ry   ÚportN)	r    r!   r"   r`   r'  r«   r*  ry   r,  r#   r$   r%   r%  r%  N  sm   € € € € € € Ø!˜'Ð";Ð=VÑWÔW€JÐWÐWÐWÑWØ�GÐ/°Ñ=Ô=€DÐ=Ð=Ð=Ñ=Ø�GÐ/°Ñ6Ô6€DÐ6Ð6Ð6Ñ6Ð6Ð6r$   r%  Úcache_knobsÚcachec                  óü   — e Zd ZU  edej                             d¦  «        ¦  «        Zded<    e	dd„ ¦  «        Z
 e	dd„ ¦  «        Z e	d	d
„ ¦  «        Z edd¦  «        Zded<    edd¦  «        Zded<   dd„ZdS )r-  ÚTRITON_HOMEz~/r`   Úhome_dirÚTRITON_DUMP_DIRc                 ó6   — t                                d¦  «        S )NÚdump©r.  Úget_triton_dirr#   r$   r%   ú<lambda>zcache_knobs.<lambda>Z  s   € Å5×CWÒCWÐX^ÑC_ÔC_€ r$   ÚTRITON_OVERRIDE_DIRc                 ó6   — t                                d¦  «        S )NÚoverrider5  r#   r$   r%   r7  zcache_knobs.<lambda>[  s   € Í5×K_ÒK_Ð`jÑKkÔKk€ r$   ÚTRITON_CACHE_DIRc                 ó6   — t                                d¦  «        S )Nr.  r5  r#   r$   r%   r7  zcache_knobs.<lambda>\  s   € ½u×?SÒ?SÐT[Ñ?\Ô?\€ r$   ÚTRITON_CACHE_MANAGERr   zenv_class[CacheManager]Úmanager_classÚTRITON_REMOTE_CACHE_BACKENDr   zenv_class[RemoteCacheBackend]Úremote_manager_classr¾   r)   r,   c                óN   — t           j                             | j        d|¦  «        S )Nz.triton)r0   r™   r½   r1  )rA   r¾   s     r%   r6  zcache_knobs.get_triton_dira  s   € ÝŒw�|Š|˜DœM¨9°gÑ>Ô>Ð>r$   N)r¾   r)   r,   r)   )r    r!   r"   r`   r0   r™   Ú
expanduserr1  r«   rn   Údump_dirÚoverride_dirÚdirrƒ   r>  r@  r6  r#   r$   r%   r-  r-  W  sÚ   € € € € € € Ø˜ ¨r¬w×/AÒ/AÀ$Ñ/GÔ/GÑHÔH€HÐHÐHÐHÑHà'Ð'Ð(9Ð;_Ð;_Ñ`Ô`€HØ+Ð+Ð,AÐCkÐCkÑlÔl€LØ
"Ð
"Ð#5Ð7\Ð7\Ñ
]Ô
]€Cà-6¨YÐ7MÈ~Ñ-^Ô-^€MÐ^Ð^Ð^Ñ^Ø:C¸)ÐDaÐcwÑ:xÔ:xÐÐxÐxÐxÑxð?ð ?ð ?ð ?ð ?ð ?r$   c                  ó¤  — e Zd ZU  ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded	<    ed
¦  «        Z	ded<    e
d¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    edd¦  «                             ¦   «         Zded<   dZded<   dS )Úcompilation_knobsÚTRITON_KERNEL_OVERRIDErt   r:  ÚTRITON_KERNEL_DUMPÚdump_irÚLLVM_EXTRACT_DI_LOCAL_VARIABLESÚ"dump_ir_extract_di_local_variablesÚTRITON_STORE_BINARY_ONLYÚstore_binary_onlyÚTRITON_ALWAYS_COMPILEÚalways_compileÚ
USE_IR_LOCrÈ   Ú
use_ir_locÚTRITON_ENABLE_ASANÚenable_asanÚTRITON_DISABLE_LINE_INFOÚdisable_line_infoÚTRITON_FRONT_END_DEBUGGINGÚfront_end_debuggingÚ"TRITON_ALLOW_NON_CONSTEXPR_GLOBALSÚallow_non_constexpr_globalsÚTRITON_INSTRUMENTATION_MODEÚ r)   Úinstrumentation_modeNz Union[CompilationListener, None]Úlistener)r    r!   r"   rt   r:  r«   rJ  rL  rN  rP  rÈ   rR  rT  rV  rX  rZ  r`   rM   r]  r^  r#   r$   r%   rG  rG  e  sl  € € € € € € Ø!˜Ð":Ñ;Ô;€HÐ;Ð;Ð;Ñ;Ø ˜Ð!5Ñ6Ô6€GÐ6Ð6Ð6Ñ6Ø3;°8Ð<]Ñ3^Ô3^Ð&Ð^Ð^Ð^Ñ^Ø"* (Ð+EÑ"FÔ"FÐÐFÐFÐFÑFØ'˜xÐ(?Ñ@Ô@€NÐ@Ð@Ð@Ñ@à)˜k¨,Ñ7Ô7€JÐ7Ð7Ð7Ñ7Ø$˜HÐ%9Ñ:Ô:€KÐ:Ð:Ð:Ñ:Ø"* (Ð+EÑ"FÔ"FÐÐFÐFÐFÑFØ$, HÐ-IÑ$JÔ$JÐÐJÐJÐJÑJØ,4¨HÐ5YÑ,ZÔ,ZÐÐZÐZÐZÑZð !( Ð(EÀrÑ JÔ J× NÒ NÑ PÔ PÐÐPÐPÐPÑPØ15€HÐ5Ð5Ð5Ñ5Ð5Ð5r$   rG  c                  ó   — e Zd Zdd„ZdS )ÚAutotuneListenerÚfnr   r(   ÚtupleÚbest_configr   Úconfigs_timingsúdict[Config, list[float]]ÚdurationúOptional[float]rç   r&   r,   r-   c               ó   — d S r/   r#   )rA   ra  r(   rc  rd  rf  rç   s          r%   ré   zAutotuneListener.__call__y  rê   r$   N)ra  r   r(   rb  rc  r   rd  re  rf  rg  rç   r&   r,   r-   rë   r#   r$   r%   r`  r`  w  rì   r$   r`  c                  ó^   — e Zd ZU  ed¦  «        Zded<    ed¦  «        Zded<   dZded<   dS )	Úautotuning_knobsÚTRITON_CACHE_AUTOTUNINGrt   r.  ÚTRITON_PRINT_AUTOTUNINGÚprintNzUnion[AutotuneListener, None]r^  )r    r!   r"   rt   r.  r«   rm  r^  r#   r$   r%   rj  rj  ~  s[   € € € € € € Ø�hÐ8Ñ9Ô9€EÐ9Ð9Ð9Ñ9Ø�hÐ8Ñ9Ô9€EÐ9Ð9Ð9Ñ9Ø.2€HÐ2Ð2Ð2Ñ2Ð2Ð2r$   rj  c                  ó   — e Zd ZdZdd„ZdS )	Ú
LaunchHookz3Hook invoked before and after kernel launching
    râ   r   r,   r-   c                ó   — d S r/   r#   )rA   râ   s     r%   ré   zLaunchHook.__call__ˆ  ó   € Øˆr$   N)râ   r   r,   r-   ©r    r!   r"   rÜ   ré   r#   r$   r%   ro  ro  „  s2   € € € € € ðð ðð ð ð ð ð r$   ro  c                  ó   — e Zd ZdZdd„ZdS )ÚInitHandleHookz}Hook invoked around kernel binary/module loading.
    module/function can be None for the *start* hook (before loading).
    ÚmodulerI   ÚfunctionúOptional[Callable]rE   r)   rä   rå   Úhashr,   r-   c                ó   — d S r/   r#   )rA   ru  rv  rE   rä   rx  s         r%   ré   zInitHandleHook.__call__‘  s	   € ð 	ˆr$   N)ru  rI   rv  rw  rE   r)   rä   rå   rx  r)   r,   r-   rr  r#   r$   r%   rt  rt  Œ  s2   € € € € € ðð ðð ð ð ð ð r$   rt  ÚFc                  ó2   — e Zd ZdZddd„Zdd
„Zdd„Zd„ ZdS )Ú	HookChainz?A chain of hooks of the same type F to be called in order.
    FÚreversedr&   c                ó"   — g | _         || _        d S r/   )Úcallsr}  )rA   r}  s     r%   rB   zHookChain.__init__£  s   € Ø ˆŒ
Ø ˆŒˆˆr$   Úfuncrz  r,   r-   c                óP   — || j         vr| j                              |¦  «         d S d S r/   )r  Úappend©rA   r€  s     r%   ÚaddzHookChain.add§  s3   € Ø�t”zÐ!Ð!ØŒJ×Ò˜dÑ#Ô#Ð#Ð#Ð#ð "Ð!r$   c                óP   — || j         v r| j                              |¦  «         d S d S r/   )r  Úremoverƒ  s     r%   r†  zHookChain.remove«  s3   € Ø�4”:ÐÐØŒJ×Ò˜dÑ#Ô#Ð#Ð#Ð#ð Ðr$   c                ó`   — | j         s| j        nt          | j        ¦  «        D ]
} ||i |¤Ž Œd S r/   )r}  r  )rA   ÚargsÚkwargsÚcalls       r%   ré   zHookChain.__call__¯  sJ   € Ø&*¤mÐM�D”J�J½À$Ä*Ñ9MÔ9Mð 	"ð 	"ˆDØˆD�$Ð!˜&Ð!Ð!Ð!Ð!ð	"ð 	"r$   Nrw   )r}  r&   )r€  rz  r,   r-   )r    r!   r"   rÜ   rB   r„  r†  ré   r#   r$   r%   r|  r|  Ÿ  sn   € € € € € ðð ð!ð !ð !ð !ð !ð$ð $ð $ð $ð$ð $ð $ð $ð"ð "ð "ð "ð "r$   r|  c                  ó’   — e Zd ZU ded<   ded<   ded<   ded<   ded	<   ded
<   ded<   ded<   ded<   ded<   ded<   ded<   ded<   dS )ÚJITHookCompileInfor)   r(   zdict[KernelParam, str]Ú	signaturer9   Údevicer-   Ú	constantsÚ	num_warpsÚnum_ctasÚ
num_stagesr&   Úenable_fp_fusionÚlaunch_cooperative_gridztuple[tuple[str, str], ...]Úextern_libsz-list[dict[tuple[int, ...], list[KernelAttr]]]ÚconfigsÚspecialization_dataÚ	is_warmupN)r    r!   r"   r«   r#   r$   r%   rŒ  rŒ  ¹  s—   € € € € € € Ø€H€H�HØ%Ð%Ð%Ñ%Ø€K€K�KØ€O€O�OØ€N€N�NØ€M€M�MØ€O€O�OØÐÐÑØ!Ð!Ð!Ñ!Ø,Ð,Ð,Ñ,Ø:Ð:Ð:Ñ:ØÐÐÑØ€O€O�O€O€Or$   rŒ  c                  ó   — e Zd Zdd„ZdS )ÚJITHookr(   r)   Úreprra  r   ÚcompilerŒ  Úis_manual_warmupr&   Úalready_compiledr,   úOptional[bool]c               ó   — d S r/   r#   )rA   r(   r›  ra  rœ  r�  rž  s          r%   ré   zJITHook.__call__Ë  rê   r$   N)r(   r)   r›  r)   ra  r   rœ  rŒ  r�  r&   rž  r&   r,   rŸ  rë   r#   r$   r%   rš  rš  É  rì   r$   rš  c                  ó   — e Zd Zd„ ZdS )ÚPipelineStagesHookc                ó   — d S r/   r#   )rA   ÚstagesÚoptionsÚlanguageÚ
capabilitys        r%   ré   zPipelineStagesHook.__call__Ò  rq  r$   Nrë   r#   r$   r%   r¢  r¢  Ð  s#   € € € € € ðð ð ð ð r$   r¢  c                  ó\  — e Zd ZU  ed¦  «        Zded<    ed¦  «                             ¦   «         Zded<    ed¦  «        Z	ded	<    e
¦   «         Zd
ed<    e
d¬¦  «        Zd
ed<    e
¦   «         Zded<    e
d¬¦  «        Zded<    e
¦   «         Zded<   dZded<   dZded<   dZded<   dS )Úruntime_knobsÚTRITON_INTERPRETrt   Ú	interpretÚTRITON_DEBUGr&   ÚdebugÚTRITON_OVERRIDE_ARCHrÈ   Úoverride_archzHookChain[LaunchHook]Úlaunch_enter_hookT)r}  Úlaunch_exit_hookzHookChain[InitHandleHook]Úkernel_load_start_hookÚkernel_load_end_hookÚkernel_unload_hookNzOptional[JITHook]Újit_cache_hookÚjit_post_compile_hookzOptional[PipelineStagesHook]Úadd_stages_inspection_hook)r    r!   r"   rt   r«  r«   rM   r­  rÈ   r¯  r|  r°  r±  r²  r³  r´  rµ  r¶  r·  r#   r$   r%   r©  r©  Ö  s8  € € € € € € Ø"˜(Ð#5Ñ6Ô6€IÐ6Ð6Ð6Ñ6ð �(˜>Ñ*Ô*×.Ò.Ñ0Ô0€EÐ0Ð0Ð0Ñ0Ø!, Ð-CÑ!DÔ!D€MÐDÐDÐDÑDà/8¨y©{¬{ÐÐ:Ð:Ð:Ñ:Ø.7¨iÀÐ.FÑ.FÔ.FÐÐFÐFÐFÑFØ8A¸	¹¼ÐÐCÐCÐCÑCØ6?°iÈÐ6NÑ6NÔ6NÐÐNÐNÐNÑNð 5>°I±K´KÐÐ?Ð?Ð?Ñ?ð )-€NÐ,Ð,Ð,Ñ,ð 04ÐÐ3Ð3Ð3Ñ3ð @DÐÐCÐCÐCÑCÐCÐCr$   r©  c                  óR   — e Zd ZU  ed¦  «        Zded<    edd¦  «        Zded<   dS )	Úlanguage_knobsÚTRITON_F32_DEFAULTrÈ   Úfp32_defaultÚTRITON_DEFAULT_FP_FUSIONTrt   Údefault_fp_fusionN)r    r!   r"   rÈ   r»  r«   rt   r½  r#   r$   r%   r¹  r¹  ï  sM   € € € € € € Ø + Ð,@Ñ AÔ A€LÐAÐAÐAÑAØ"* (Ð+EÀtÑ"LÔ"LÐÐLÐLÐLÑLÐLÐLr$   r¹  c                  óp  — e Zd ZU  ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    e	d¦  «        Z
ded	<    e	d
¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    e	d¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<   dS )Únvidia_knobsÚ	cuobjdumpr°   ÚnvdisasmÚptxaszptxas-blackwellÚptxas_blackwellÚNVPTX_ENABLE_DUMPrt   Ú
dump_nvptxÚDISABLE_PTXAS_OPTÚdisable_ptxas_optÚPTXAS_OPTIONSrÈ   Úptxas_optionsÚTRITON_MOCK_PTX_VERSIONÚmock_ptx_versionÚTRITON_DUMP_PTXAS_LOGÚdump_ptxas_logÚTRITON_LIBDEVICE_PATHÚlibdevice_pathÚTRITON_LIBCUDA_PATHÚlibcuda_pathN)r    r!   r"   r°   rÀ  r«   rÁ  rÂ  rÃ  rt   rÅ  rÇ  rÈ   rÉ  rË  rÍ  rÏ  rÑ  r#   r$   r%   r¿  r¿  ô  sD  € € € € € € Ø!0 °Ñ!=Ô!=€IÐ=Ð=Ð=Ñ=Ø / °
Ñ ;Ô ;€HÐ;Ð;Ð;Ñ;Ø,˜_¨WÑ5Ô5€EÐ5Ð5Ð5Ñ5Ø'6 Ð7HÑ'IÔ'I€OÐIÐIÐIÑIà#˜8Ð$7Ñ8Ô8€JÐ8Ð8Ð8Ñ8Ø"* (Ð+>Ñ"?Ô"?ÐÐ?Ð?Ð?Ñ?Ø!, ¨_Ñ!=Ô!=€MÐ=Ð=Ð=Ñ=Ø$/ KÐ0IÑ$JÔ$JÐÐJÐJÐJÑJØ'˜xÐ(?Ñ@Ô@€NÐ@Ð@Ð@Ñ@à"- +Ð.EÑ"FÔ"F€NÐFÐFÐFÑFØ + Ð,AÑ BÔ B€LÐBÐBÐBÑBÐBÐBr$   r¿  c                  óØ  — e Zd ZU  edd¦  «        Zded<    edd¦  «        Zded<    edd¦  «        Zded	<    ed
¦  «        Zded<    e	d¦  «        Z
ded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    ed¦  «        Zded<    e	d¦  «        Zded<    e	d¦  «        Zded<    ed d¦  «        Zded!<   d"S )#Ú	amd_knobsÚAMDGCN_USE_BUFFER_OPSTrt   Úuse_buffer_opsÚAMDGCN_USE_BUFFER_ATOMICSÚuse_buffer_atomicsÚ!AMDGCN_ANALYZE_SMALL_TENSOR_RANGEFÚ%buffer_ops_analyze_small_tensor_rangeÚAMDGCN_ENABLE_DUMPÚdump_amdgcnÚTRITON_LIBHIP_PATHrÈ   Úlibhip_pathÚTRITON_HIP_USE_BLOCK_PINGPONGrÍ   Úuse_block_pingpongÚ"TRITON_HIP_USE_IN_THREAD_TRANSPOSEÚuse_in_thread_transposeÚTRITON_HIP_USE_ASYNC_COPYÚuse_async_copyÚTRITON_HIP_USE_COEXEC_SCHEDULERÚuse_coexec_schedulerÚ TRITON_HIP_USE_EXPERT_SCHEDULINGÚuse_expert_schedulingÚAMDGCN_SCALARIZE_PACKED_FOPSÚscalarize_packed_fopsÚTRITON_DUMP_MIRÚdump_mirÚTRITON_SWAP_MIRÚswap_mirÚTRITON_SWAP_MIR_ENABLE_MISCHEDÚswap_mir_enable_mischedN)r    r!   r"   rt   rÕ  r«   r×  rÙ  rÛ  rÈ   rÝ  rÍ   rß  rá  rã  rå  rç  ré  rë  rí  rï  r#   r$   r%   rÓ  rÓ    s°  € € € € € € Ø'˜xÐ(?ÀÑFÔF€NÐFÐFÐFÑFà#+ 8Ð,GÈÑ#NÔ#NÐÐNÐNÐNÑNà6>°hÐ?bÐdiÑ6jÔ6jÐ)ÐjÐjÐjÑjØ$˜HÐ%9Ñ:Ô:€KÐ:Ð:Ð:Ñ:Ø*˜{Ð+?Ñ@Ô@€KÐ@Ð@Ð@Ñ@ð (4 |Ð4SÑ'TÔ'TÐÐTÐTÐTÑTØ,8¨LÐ9]Ñ,^Ô,^ÐÐ^Ð^Ð^Ñ^Ø#/ <Ð0KÑ#LÔ#L€NÐLÐLÐLÑLØ)5¨Ð6WÑ)XÔ)XÐÐXÐXÐXÑXØ*6¨,Ð7YÑ*ZÔ*ZÐÐZÐZÐZÑZà&. hÐ/MÑ&NÔ&NÐÐNÐNÐNÑNð (˜KÐ(9Ñ:Ô:€HÐ:Ð:Ð:Ñ:à'˜KÐ(9Ñ:Ô:€HÐ:Ð:Ð:Ñ:à(0¨Ð1QÐSXÑ(YÔ(YÐÐYÐYÐYÑYÐYÐYr$   rÓ  c                  ó&  — e Zd ZU  edd¦  «        Zded<    ed e ej	        e
¦  «        j                             ¦   «         dz  dz  dz  d	z  ¦  «        ¦  «        Zd
ed<    ed e ej	        e
¦  «        j                             ¦   «         dz  dz  dz  dz  ¦  «        ¦  «        Zd
ed<    ed¦  «        Zded<    ed¦  «        Zded<    edd¦  «        Zded<    edd¦  «        Zded<    edd¦  «        Zded<    edd¦  «        Zded<   dS ) Úproton_knobsÚTRITON_PROTON_DISABLEFrt   ÚdisableÚTRITON_CUPTI_LIB_PATHr´   rµ   ÚlibÚcuptir`   Úcupti_lib_dirÚTRITON_CUPTI_LIB_BLACKWELL_PATHzcupti-blackwellÚcupti_lib_blackwell_dirÚ#TRITON_ROCPROFILER_SDK_INCLUDE_PATHrÈ   Úrocprofiler_sdk_include_pathÚTRITON_ROCPROFILER_SDK_LIB_PATHÚrocprofiler_sdk_lib_pathÚTRITON_PROFILE_BUFFER_SIZEi   ry   Úprofile_buffer_sizeÚ!TRITON_PROFILE_METRIC_BUFFER_SIZEÚprofile_metric_buffer_sizeÚTRITON_ENABLE_NVTXTÚenable_nvtxÚTRITON_ENABLE_HW_TRACEÚenable_hw_traceN)r    r!   r"   rt   ró  r«   r`   r)   ÚpathlibÚPathr¿   ÚparentÚabsoluter÷  rù  rÈ   rû  rý  ry   rÿ  r  r  r  r#   r$   r%   rñ  rñ    sÃ  € € € € € € Ø ˜Ð!8¸%Ñ@Ô@€GÐ@Ð@Ð@Ñ@Ø$˜WØØˆˆLˆGŒL˜Ñ"Ô"Ô)×2Ò2Ñ4Ô4°zÑAÀHÑLÈuÑTÐW^Ñ^Ñ_Ô_ñaô a€Mð að að añ að (/ wØ)ØˆˆLˆGŒL˜Ñ"Ô"Ô)×2Ò2Ñ4Ô4°zÑAÀHÑLÈuÑTÐWhÑhÑiÔiñ(kô (kÐð kð kð kñ kð 1<°Ð<aÑ0bÔ0bÐ ÐbÐbÐbÑbØ,7¨KÐ8YÑ,ZÔ,ZÐÐZÐZÐZÑZØ#* 7Ð+GÐIYÑ#ZÔ#ZÐÐZÐZÐZÑZØ*1¨'Ð2UÐWgÑ*hÔ*hÐÐhÐhÐhÑhØ$˜HÐ%9¸4Ñ@Ô@€KÐ@Ð@Ð@Ñ@ð$ !) Ð)AÀ5Ñ IÔ I€OÐIÐIÐIÑIÐIÐIr$   rñ  c                 ó´   — t          d¦  «                             ¦   «         t          _        t	          dd¦  «                             ¦   «         t
          _        d S )Nr¬  r[  r\  )rt   rM   Úruntimer­  r`   Úcompilationr]  r#   r$   r%   Úrefresh_knobsr  K  sB   € Ý˜^Ñ,Ô,×0Ò0Ñ2Ô2…G„MÝ'.Ð/LÈbÑ'QÔ'Q×'UÒ'UÑ'WÔ'W…KÔ$Ð$Ð$r$   )r(   r)   r*   r+   r,   r-   )r3   r   r,   r4   )fÚ
__future__r   r­   r‘   r0   r¢   rŸ   r»   r  Údataclassesr   Ú
contextlibr   Útypingr   r   r   r	   r
   r   r   r   r   r   r   r   Útriton._C.libtritonr   r   Úruntime.cacher   r   Úruntime.jitr   r   r   Úruntime.autotunerr   Úcompiler.compilerr   r   r   r   Úenvr'   r«   r2   r;   r<   r=   rU   rN   r?   r)   r`   rn   r&   rt   r9   ry   r�   rƒ   r˜   r°   rÈ   rÍ   rÑ   rß   rí   rî   r  r  r%  r-  rG  r`  rj  ro  rt  rz  r|  ÚlistÚ
KernelAttrrŒ  rš  r¢  r©  r¹  r¿  rÓ  rñ  ÚbuildÚredisr.  r  Ú
autotuningr  r¦  rµ   ÚamdÚprotonr  r#   r$   r%   ú<module>r     sÞ  ðØ "Ð "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ø Ð Ð Ð Ø 	€	€	€	Ø 	€	€	€	Ø Ð Ð Ð Ø Ð Ð Ð Ø €€€à !Ð !Ð !Ð !Ð !Ð !Ø %Ð %Ð %Ð %Ð %Ð %Ø ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~Ð ~à 3Ð 3Ð 3Ð 3Ð 3Ð 3Ð 3Ð 3àð AØ?Ð?Ð?Ð?Ð?Ð?Ð?Ð?ØFÐFÐFÐFÐFÐFÐFÐFÐFÐFØ)Ð)Ð)Ð)Ð)Ð)Ø@Ð@Ð@Ð@Ð@Ð@Ð@Ð@Ð@Ð@ð	ð 	ð 	ð 	ð 	ñ 	ô 	ð 	ð 
€c�e„e€à€Ð Ð Ð Ñ ðð ð ð ðð ð ð ð& ˆ'�)Ñ
Ô
€Ø
ˆ'�)Ñ
Ô
€àˆ6‰8Œ8€ð"ð "ð "ð "ð "ˆw�w Ð'Ô(ñ "ô "ð "ðD.ð .ð .ð .ð .ˆh�s˜C�xÔ ñ .ô .ð .ð
ð 
ð 
ð 
ð 
˜x¨¨S¨Ô1ñ 
ô 
ð 
ð3ð 3ð 3ð 3ð 3ˆx˜˜d˜
Ô#ñ 3ô 3ð 3ðYð Yð Yð Yð Yˆh�s˜C�xÔ ñ Yô Yð Yð  ˆG�KÑ Ô €	ð*ð *ð *ð *ð *�˜	Ô" H¨X°d¸9´oÔ-FÈÐQUÐV_ÔQ`ÔHaÐ-aÔ$bñ *ô *ð *ð, ðð ð ð ð ñ ô ñ „ðð"9ð 9ð 9ð 9ð 9�h˜s J˜Ô/ñ 9ô 9ð 9ð8 ð  ð  ð  ð  �(˜8 Cœ=¨(°3¬-Ð7Ô8ñ  ô  ð  ð+ð +ð +ð +ð +�8ñ +ô +ð +ð €�$ÐÑÔðQð Qð Qð Qð Qñ Qô Qñ ÔðQð4ð ð ð ð ˜(ñ ô ð ð ˆW�\¨Ð6Ñ6Ô6€
ð'&ð '&ð '&ð '&ð '&ñ '&ô '&ð '&ðTð ð ð ð �ñ ô ð ð\ð \ð \ð \ð \�*ñ \ô \ð \ð7ð 7ð 7ð 7ð 7�*ñ 7ô 7ð 7ð Ð Ð Ñ ð?ð ?ð ?ð ?ð ?�*ñ ?ô ?ð ?ð6ð 6ð 6ð 6ð 6˜
ñ 6ô 6ð 6ð$ð ð ð ð �xñ ô ð ð3ð 3ð 3ð 3ð 3�zñ 3ô 3ð 3ðð ð ð ð �ñ ô ð ðð ð ð ð �Xñ ô ð ð  €GˆC�xÐ Ñ Ô €ð"ð "ð "ð "ð "�˜”
ñ "ô "ð "ð. �%˜˜S˜”/Ô"€
ðð ð ð ð ˜ñ ô ð ð ð ð ð ð ˆhñ ô ð ðð ð ð ð ˜ñ ô ð ðDð Dð Dð Dð D�Jñ Dô Dð Dð2Mð Mð Mð Mð M�Zñ Mô Mð Mð
Cð Cð Cð Cð C�:ñ Cô Cð Cð Zð Zð Zð Zð Z�
ñ Zô Zð Zð4Jð Jð Jð Jð J�:ñ Jô Jð JðB 	ˆ‰Œ€Øˆ‰Œ€Øˆ‰Œ€ØÐÑ!Ô!€ØÐÑÔ€
Ø
ˆ-‰/Œ/€Øˆ>ÑÔ€Ø	ˆ‰Œ€Ø€i�k„k€Ø	ˆ‰Œ€ðXð Xð Xð Xð Xr$   