Ë
    ÍÍ:j¼8  ã            	      óð  — 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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 erddlmZmZ ddlmZmZ dd	lmZm Z m!Z!  G d
„ d«      Z" e"«       Z#dZ$de%d<   dOd„Z&dPd„Z'dQd„Z( ed«      Z) ed«      Z* G d„ dee)e*f   «      Z+ G d„ de+e,e,f   «      Z- G d„ de+e.e.f   «      Z/ G d„ de+e0e0f   «      Z1 G d„ dee*e)f   e+ee*   ee)   f   «      Z2 ed«      Z3 G d„ d ee3   e2ee3   ee3   f   «      Z4e G d!„ d"«      «       Z5 G d#„ d$e+e,e5f   «      Z6 G d%„ d&e2e,e,f   e-«      Z7 G d'„ d(e2e.e.f   e/«      Z8 ed¬)«       G d*„ d+«      «       Z9 G d,„ d-e«      Z: ed.d/¬0«      Z; G d1„ d/«      Z< G d2„ d3e«      Z= G d4„ d5e<«      Z> G d6„ d7e<«      Z?d8e%d9<    G d:„ d8e<«      Z@ G d;„ d<e<«      ZA G d=„ d>e<«      ZB G d?„ d@e«      ZCeDee,e0f      ZE G dA„ dBe«      ZF G dC„ dDe«      ZG G dE„ dFe<«      ZH G dG„ dHe<«      ZI G dI„ dJe<«      ZJ G dK„ dLe<«      ZK G dM„ dNe<«      ZL e>«       ZM e?«       ZN e@«       ZO eA«       ZP eB«       ZQ eH«       ZR eI«       ZS eJ«       ZT eK«       ZU eL«       ZVy)Ré    )ÚannotationsN)Ú	dataclass)Úcontextmanager)ÚcastÚAnyÚCallableÚ	GeneratorÚGenericÚOptionalÚProtocolÚTypeÚTypeVarÚ	TypedDictÚTYPE_CHECKINGÚUnioné   )ÚCacheManagerÚRemoteCacheBackend)ÚJitFunctionInfoÚKernelParam)Ú	ASTSourceÚLazyDictÚIRSourcec                  ó   — e Zd Zy)ÚEnvN©Ú__name__Ú
__module__Ú__qualname__© ó    úa/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/triton/knobs.pyr   r      ó   „ Ør!   r   TÚboolÚpropagate_envc                óT   — t        j                  | «      }|�|j                  «       S |S ©N)ÚosÚgetenvÚstrip)ÚkeyÚress     r"   r)   r)      s$   € Ü
�)‰)�C‹.€CØ˜/ˆ3�9‰9‹;Ð2¨sÐ2r!   c                ó†   — t         sy |�|t        j                  | <   y | t        j                  v rt        j                  | = y y r'   )r%   r(   Úenviron)r+   Úvalues     r"   Úsetenvr0   !   s7   € ÝØàÐØŒ�
‰
�3ŠØ	”—
‘
Ñ	Ü�J‰J�s‰Oð 
r!   c                ó~   — | €yt        | «      }|t        u r| rdfS dfS |t        u r| fS |t        u rt        | «      fS y )Nr'   Ú1Ú0)Útyper$   ÚstrÚint)ÚvalÚts     r"   Útoenvr9   +   sQ   € Ø
€{ØäˆS‹	€AØŒD�yÙ�Ð&Ð& Ð&Ð&àŒC�xØˆwˆàŒC�xÜ�C“ˆ|Ðàr!   ÚSetTypeÚGetTypec                  ó^   — e Zd Zdd„Zdd„Zdd„Zedd„«       Zdd„Zdd„Z	dd„Z
dd„Zdd	„Zy
)Úenv_basec                óN   ‡— || _         t        ‰«      r‰| _        y ˆfd„| _        y )Nc                 ó   •— ‰ S r'   r    )Údefaults   €r"   ú<lambda>z#env_base.__init__.<locals>.<lambda>F   s   ø€ ÐX_€ r!   )r+   Úcallabler@   )Úselfr+   r@   s     `r"   Ú__init__zenv_base.__init__D   s   ø€ ØˆŒÜ9AÀ'Ô9J¨gˆ�ÓP_ˆ�r!   c                ó   — || _         y r'   )Úname)rC   ÚobjclassrF   s      r"   Ú__set_name__zenv_base.__set_name__H   s	   € Øˆ�	r!   c                óÖ   — |€t        dt        | «      › d�«      ‚| j                  |j                  v r(| j	                  |j                  | j                     «      S | j                  «       S )NzCannot access z on non-instance)ÚAttributeErrorr4   rF   Ú__dict__Ú	transformÚget)rC   ÚobjrG   s      r"   Ú__get__zenv_base.__get__K   sX   € Øˆ;Ü  >´$°t³*°Ð=MÐ!NÓOÐOà�9‰9˜Ÿ™Ñ$Ø—>‘> #§,¡,¨t¯y©yÑ"9Ó:Ð:à—8‘8“:Ðr!   c                ó,   — t        | j                  «      S r'   )r)   r+   ©rC   s    r"   Úenv_valzenv_base.env_valT   s   € ä�d—h‘hÓÐr!   c                ó†   — | j                   }| j                  |€| j                  «       «      S | j                  |«      «      S r'   )rR   rL   r@   Úfrom_env)rC   Úenvs     r"   rM   zenv_base.getX   s5   € Ø�l‰lˆØ�~‰~°°˜dŸl™l›nÓTÐTÀÇÁÈsÓASÓTÐTr!   c                óò   — t        |t        «      r'|j                  j                  | j                  d «       y ||j                  | j                  <   t        |«      x}rt        | j                  |d   «       y y )Nr   )Ú
isinstancer   rK   ÚpoprF   r9   r0   r+   )rC   rN   r/   rR   s       r"   Ú__set__zenv_base.__set__\   s\   € Ü�eœSÔ!Ø�L‰L×Ñ˜TŸY™Y¨Õ-à&+ˆC�L‰L˜Ÿ™Ñ#Ü ›,Ð&ˆwÐ&Ü�t—x‘x ¨¡Õ,ð 'r!   c                óP   — |j                   j                  | j                  d «       y r'   )rK   rX   rF   )rC   rN   s     r"   Ú
__delete__zenv_base.__delete__d   s   € Ø�‰×Ñ˜Ÿ™ DÕ)r!   c                ó"   — t        t        |«      S r'   )r   r;   ©rC   r7   s     r"   rL   zenv_base.transformg   s   € ô ”G˜SÓ!Ð!r!   c                ó   — t        «       ‚r'   )ÚNotImplementedErrorr]   s     r"   rT   zenv_base.from_envl   s   € Ü!Ó#Ð#r!   N)r+   r5   r@   z%Union[SetType, Callable[[], SetType]]ÚreturnÚNone)rG   zType[object]rF   r5   r`   ra   )rN   zOptional[object]rG   zOptional[Type[object]]r`   r;   )r`   z
str | None)r`   r;   )rN   Úobjectr/   zUnion[SetType, Env]r`   ra   )rN   rb   r`   ra   )r7   r:   r`   r;   )r7   r5   r`   r:   )r   r   r   rD   rH   rO   ÚpropertyrR   rM   rY   r[   rL   rT   r    r!   r"   r=   r=   B   s@   „ ó`óóð ò ó ð óUó-ó*ó"ô
$r!   r=   c                  ó   — e Zd Zdd„Zy)Úenv_strc                ó   — |S r'   r    r]   s     r"   rT   zenv_str.from_envr   ó   € Øˆ
r!   N©r7   r5   r`   r5   )r   r   r   rT   r    r!   r"   re   re   p   s   „ ôr!   re   c                  ó*   ‡ — e Zd Zddˆ fd„Zdd„Zˆ xZS )Úenv_boolc                ó&   •— t         ‰| �  ||«       y r'   ©ÚsuperrD   ©rC   r+   r@   Ú	__class__s      €r"   rD   zenv_bool.__init__x   ó   ø€ Ü‰Ñ˜˜gÕ&r!   c                ó&   — |j                  «       dv S )N)r2   ÚtrueÚyesÚonÚy)Úlowerr]   s     r"   rT   zenv_bool.from_env{   s   € Ø�y‰y‹{Ð=Ð=Ð=r!   )F)r+   r5   r@   zUnion[bool, Callable[[], bool]]r`   ra   )r7   r5   r`   r$   ©r   r   r   rD   rT   Ú__classcell__©ro   s   @r"   rj   rj   v   s   ø„ ö'÷>r!   rj   c                  ó*   ‡ — e Zd Zddˆ fd„Zdd„Zˆ xZS )Úenv_intc                ó&   •— t         ‰| �  ||«       y r'   rl   rn   s      €r"   rD   zenv_int.__init__�   rp   r!   c                óv   — 	 t        |«      S # t        $ r"}t        d| j                  › d|› d�«      |‚d }~ww xY w)NzUnable to use ú=z: expected int)r6   Ú
ValueErrorÚRuntimeErrorr+   )rC   r7   Úexcs      r"   rT   zenv_int.from_env„   sD   € ð	YÜ�s“8ˆOøÜò 	YÜ °·±¨z¸¸3¸%¸~ÐNÓOÐUXÐXûð	Yús   ‚
 �	8–3³8)r   )r+   r5   r@   zUnion[int, Callable[[], int]]r`   ra   )r7   r5   r`   r6   rw   ry   s   @r"   r{   r{      s   ø„ ö'÷Yr!   r{   c                  ó    ‡ — e Zd Zdˆ fd„Zˆ xZS )Úenv_opt_basec                ó&   •— t         ‰| �  |d «       y r'   rl   )rC   r+   ro   s     €r"   rD   zenv_opt_base.__init__�   s   ø€ Ü‰Ñ˜˜dÕ#r!   )r+   r5   r`   ra   )r   r   r   rD   rx   ry   s   @r"   rƒ   rƒ   ‹   s   ø„ ÷$ñ $r!   rƒ   Ú	ClassTypec                  ó(   ‡ — e Zd Zdˆ fd„Zdd„Zˆ xZS )Ú	env_classc                ó2   •— t         ‰| �  |«       || _        y r'   )rm   rD   r4   )rC   r+   r4   ro   s      €r"   rD   zenv_class.__init__–   s   ø€ Ü‰Ñ˜Ôàˆ�	r!   c           	     óŠ  ‡ — |j                  dd«      }t        |«      dk7  rt        d‰ j                  › d|› d�«      ‚t	        t        j                  |d   «      |d   «      }t        ˆ fd„|j                  «       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  óP   •K  — | ]  }|j                   ‰j                  k(  –— Œ y ­wr'   )r   r4   )Ú.0ÚcrC   s     €r"   ú	<genexpr>z%env_class.from_env.<locals>.<genexpr>¡   s   øè ø€ Ò?°�A—J‘J $§)¡)Õ+Ñ?ùs   ƒ#&zUnable to use 'z' from z: not of type 'ú')ÚsplitÚlenr€   r+   ÚgetattrÚ	importlibÚimport_moduleÚanyÚmror4   r   r   r…   )rC   r7   ÚcompsÚclss   `   r"   rT   zenv_class.from_env›   s®   ø€ Ø—	‘	˜#˜qÓ!ˆÜˆu‹:˜Š?Ü °·±°
¸#¸c¸UÐBbÐcÓdÐdÜ”i×-Ñ-¨e°A©hÓ7¸¸q¹ÓBˆäÓ?°S·W±W³YÔ?Ô@Ü °°°W¸T¿X¹X¸JÀoÐVZ×V_ÑV_ÐU`Ð`aÐbÓcÐcä”Dœ‘O SÓ)Ð)r!   )r+   r5   r4   r5   r`   ra   )r7   r5   r`   zType[ClassType]rw   ry   s   @r"   r‡   r‡   ”   s   ø„ õ÷
	*r!   r‡   c                  ó4   — e Zd ZU ded<   ded<   edd„«       Zy)Ú
NvidiaToolr5   ÚpathÚversionc                ó0  — 	 t        j                  | dgt         j                  ¬«      }|€y t        j                  d|j                  d«      t        j                  ¬«      }|€y t        | |j                  d«      «      S # t         j                  $ r Y y w xY w)Nz	--version)Ústderrz.*release (\d+\.\d+).*zutf-8)Úflagsr   )
Ú
subprocessÚcheck_outputÚSTDOUTÚreÚsearchÚdecodeÚ	MULTILINEr›   ÚgroupÚCalledProcessError)rœ   Úresultr�   s      r"   Ú	from_pathzNvidiaTool.from_path¬   s…   € ð		Ü×,Ñ,¨d°KÐ-@Ì×IZÑIZÔ[ˆFØˆ~ØÜ—i‘iÐ 9¸6¿=¹=ÈÓ;QÔY[×YeÑYeÔfˆGØˆØÜ˜d G§M¡M°!Ó$4Ó5Ð5øÜ×,Ñ,ò 	Ùð	ús   ‚)A? ¬7A? Á$A? Á?BÂBN)rœ   r5   r`   zOptional[NvidiaTool])r   r   r   Ú__annotations__Ústaticmethodr«   r    r!   r"   r›   r›   §   s   … à
ƒIØƒLàò
ó ñ
r!   r›   c                  ó0   ‡ — e Zd Zdˆ fd„Zdd„Zdd„Zˆ xZS )Úenv_nvidia_toolc                ó�   •‡ — |t        j                  d«      z  }|‰ _        t        ‰‰ �  d|j                  «       › d�ˆ fd„«       y )NÚEXEÚTRITON_Ú_PATHc                 ó    •— t         j                  j                  t         j                  j                  t        «      ddd‰ j
                  «      S )NÚbackendsÚnvidiaÚbin)r(   rœ   ÚjoinÚdirnameÚ__file__ÚbinaryrQ   s   €r"   rA   z*env_nvidia_tool.__init__.<locals>.<lambda>¿   s5   ø€ Ä"Ç'Á'Ç,Á,Ü�G‰G�O‰OœHÓ%ØØØØ�K‰KóC
€ r!   )Ú	sysconfigÚget_config_varr»   rm   rD   Úupper)rC   r»   ro   s   ` €r"   rD   zenv_nvidia_tool.__init__¼   sC   ù€ Ø”)×*Ñ*¨5Ó1Ñ1ˆØˆŒÜ‰Ñ˜7 6§<¡<£>Ð"2°%Ð8ó ;
õ 	r!   c                óè   — || j                  «       g}|D ]D  }|r$t        j                  |t        j                  «      sŒ*t        j                  |«      x}sŒB|c S  t        d| j                  › �«      ‚)NzCannot find )r@   r(   ÚaccessÚX_OKr›   r«   r€   r»   )rC   rœ   ÚpathsÚtools       r"   rL   zenv_nvidia_tool.transformÇ   sr   € àð �L‰L‹Nð	
ˆð ò 	ˆDÙœrŸy™y¨¬r¯w©wÔ7ØÜ!×+Ñ+¨DÓ1Ð1ˆtÑ1Ø’ð		ô ˜\¨$¯+©+¨Ð7Ó8Ð8r!   c                ó   — |S r'   r    r]   s     r"   rT   zenv_nvidia_tool.from_envÖ   rg   r!   )r»   r5   r`   ra   )rœ   r5   r`   r›   rh   )r   r   r   rD   rL   rT   rx   ry   s   @r"   r¯   r¯   º   s   ø„ õ	ó9÷r!   r¯   c                  ó   — e Zd Zy)Úenv_opt_strNr   r    r!   r"   rÆ   rÆ   Û   r#   r!   rÆ   c                  ó   — e Zd Zy)Úenv_opt_boolNr   r    r!   r"   rÈ   rÈ   ß   r#   r!   rÈ   )Úfrozenc                  óT   — e Zd ZU dZded<   ded<   ded<   ed
d„«       Zed
d„«       Zy	)ÚCompileTimeszm
    Model holding timing information for an invocation of the compiler.

    All times in microseconds.
    r6   Úir_initializationzlist[tuple[str, int]]Úlowering_stagesÚstore_resultsc                ó:   — t        d„ | j                  D «       «      S )Nc              3  ó&   K  — | ]	  }|d    –— Œ y­w)r   Nr    )r�   Ústages     r"   r�   z.CompileTimes.total_lowering.<locals>.<genexpr>÷   s   è ø€ Ò? �E˜!•HÑ?ùs   ‚)ÚsumrÍ   rQ   s    r"   Útotal_loweringzCompileTimes.total_loweringõ   s   € äÑ?¨$×*>Ñ*>Ô?Ó@Ð@r!   c                óN   — | j                   | j                  z   | j                  z   S r'   )rÌ   rÓ   rÎ   rQ   s    r"   ÚtotalzCompileTimes.totalù   s$   € à×%Ñ%¨×(;Ñ(;Ñ;¸d×>PÑ>PÑPÐPr!   N)r`   r6   )r   r   r   Ú__doc__r¬   rc   rÓ   rÕ   r    r!   r"   rË   rË   ã   sI   … ñð Óð +Ó*ð ÓàòAó ðAð òQó ñQr!   rË   c                  ó    — e Zd Z	 	 	 	 	 	 dd„Zy)ÚCompilationListenerc                ó   — y r'   r    )rC   ÚsrcÚmetadataÚmetadata_groupÚtimesÚ	cache_hits         r"   Ú__call__zCompilationListener.__call__   ó   € àr!   N)rÚ   zUnion[ASTSource, IRSource]rÛ   údict[str, Any]rÜ   zdict[str, str]rÝ   rË   rÞ   r$   r`   ra   ©r   r   r   rß   r    r!   r"   rØ   rØ   þ   s   „ ðØ$ðØ15ðØ:>ôr!   rØ   Ú
knobs_typeÚ
base_knobs)Úboundc                  óR   — e Zd Zedd„«       Zedd„«       Zd	d„Zd	d„Zed
d„«       Z	y)rä   c                óœ   — t        | «      j                  j                  «       D ��ci c]  \  }}t        |t        «      r||“Œ c}}S c c}}w r'   )r4   rK   ÚitemsrW   r=   )rC   ÚkÚvs      r"   Úknob_descriptorszbase_knobs.knob_descriptors
  sK   € ô
 ˜T›
×+Ñ+×1Ñ1Ó3÷
ñ ��1Ü˜!œXÔ&ð ˆq‰Dó
ð 	
ùó 
s   §Ac                ór   — | j                   j                  «       D �ci c]  }|t        | |«      “Œ c}S c c}w r'   )rë   Úkeysr“   )rC   ré   s     r"   Úknobszbase_knobs.knobs  s1   € à-1×-BÑ-B×-GÑ-GÓ-IÖJ¨�”7˜4 Ó#Ñ#ÒJÐJùÒJs   �4c                óp   —  t        | «      «       }|j                  j                  | j                  «       |S r'   )r4   rK   Úupdate)rC   r,   s     r"   Úcopyzbase_knobs.copy  s*   € ØŒd�4‹j‹lˆØ�‰×Ñ˜DŸM™MÔ*Øˆ
r!   c                ó\   — | j                   j                  «       D ]  }t        | |«       Œ | S r'   )rë   rí   Údelattr)rC   Úknobs     r"   Úresetzbase_knobs.reset  s/   € Ø×)Ñ)×.Ñ.Ó0ò 	 ˆDÜ�D˜$Õð	 àˆr!   c              #  óè  K  — 	 | j                   j                  «       D �ci c]  }|j                  |j                  “Œ }}t	        | j
                  «      }d –— | j
                  j                  «        | j
                  j                  |«       |j                  «       D ]?  \  }}|�|t        j                  |<   Œ|t        j                  v sŒ/t        j                  |= ŒA y c c}w # | j
                  j                  «        | j
                  j                  «       j                  «       D ]?  \  }}|�|t        j                  |<   Œ|t        j                  v sŒ/t        j                  |= ŒA w xY w­wr'   )rë   Úvaluesr+   rR   ÚdictrK   Úclearrð   rè   r(   r.   )rC   rô   Úinitial_envÚorigré   rê   s         r"   Úscopezbase_knobs.scope!  s   è ø€ ð	&Ø>B×>SÑ>S×>ZÑ>ZÓ>\Ö]°d˜4Ÿ8™8 T§\¡\Ñ1Ð]ˆKÐ]Ü˜Ÿ™Ó&ˆDÛà�M‰M×ÑÔ!Ø�M‰M× Ñ  Ô&à#×)Ñ)Ó+ò &‘��1Ø�=Ø$%”B—J‘J˜q’MØœ"Ÿ*™*’_ÜŸ
™
 1™ñ	&ùò ^øð �M‰M×ÑÔ!Ø�M‰M× Ñ  Ô&à#×)Ñ)Ó+ò &‘��1Ø�=Ø$%”B—J‘J˜q’MØœ"Ÿ*™*’_ÜŸ
™
 1™ñ	&üs:   ‚E2„C&  C!¾C& ÁA3E2ÃE2Ã!C& Ã&A4E/ÅE/Å/E2N)r`   zdict[str, env_base])r`   rá   )rC   rã   r`   rã   )r`   zGenerator[None, None, None])
r   r   r   rc   rë   rî   rñ   rõ   r   rü   r    r!   r"   rä   rä     sH   „ àò
ó ð
ð òKó ðKóó
ð
 ò&ó ñ&r!   c                  ó   — e Zd Z	 	 	 	 dd„Zy)Ú	BuildImplc                ó   — y r'   r    )rC   rF   rÚ   ÚsrcdirÚlibrary_dirsÚinclude_dirsÚ	librariess          r"   rß   zBuildImpl.__call__4  rà   r!   N)rF   r5   rÚ   r5   r   r5   r  ú	list[str]r  r  r  r  r`   r5   râ   r    r!   r"   rþ   rþ   2  s   „ ðØ%ðØ-0ô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y	)Úbuild_knobsz<Configuration controlling how the native compiler is invokedÚCCrÆ   ÚccÚTRITON_CUDACRT_PATHÚcudacrt_pathÚTRITON_CUDART_PATHÚcudart_pathNzOptional[BuildImpl]Úimplc                ó^   — | j                   | j                  fD �ch c]  }|€Œ|’Œ	 c}S c c}w r'   )r
  r  )rC   rœ   s     r"   Úbackend_dirszbuild_knobs.backend_dirsB  s,   € à"&×"3Ñ"3°T×5EÑ5EÐ!FÖ[˜È$ÑJZ’Ò[Ð[ùÒ[s   ›*£*)r`   zset[str])r   r   r   rÖ   rÆ   r  r¬   r
  r  r  rc   r  r    r!   r"   r  r  9  sS   … ÙFÙ! $Ó'€BˆÓ'á +Ð,AÓ B€L�+ÓBÙ*Ð+?Ó@€K�Ó@à $€DÐ
Ó$àò\ó ñ\r!   r  c                  ób   — 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<   y)Úredis_knobsÚTRITON_REDIS_KEY_FORMATztriton:{key}:{filename}re   Ú
key_formatÚTRITON_REDIS_HOSTÚ	localhostÚhostÚTRITON_REDIS_PORTië  r{   ÚportN)	r   r   r   re   r  r¬   r  r{   r  r    r!   r"   r  r  G  s9   … Ù!Ð";Ð=VÓW€J�ÓWÙÐ/°Ó=€Dˆ'Ó=ÙÐ/°Ó6€Dˆ'Ô6r!   r  Úcache_knobsÚcachec                  óÆ   — 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d„ «      Zded<    e	dd«      Z
ded<    e	dd«      Zded<   dd„Zy)r  ÚTRITON_HOMEc                 ó@   — t         j                  j                  d«      S )Nz~/)r(   rœ   Ú
expanduserr    r!   r"   rA   zcache_knobs.<lambda>Q  s   € ´r·w±w×7IÑ7IÈ$Ó7O€ r!   re   Úhome_dirÚTRITON_DUMP_DIRc                 ó,   — t         j                  d«      S )NÚdump©r  Úget_triton_dirr    r!   r"   rA   zcache_knobs.<lambda>S  s   € ¼5×;OÑ;OÐPVÓ;W€ r!   Údump_dirÚTRITON_OVERRIDE_DIRc                 ó,   — t         j                  d«      S )NÚoverrider#  r    r!   r"   rA   zcache_knobs.<lambda>T  s   € Ä5×CWÑCWÐXbÓCc€ r!   Úoverride_dirÚTRITON_CACHE_DIRc                 ó,   — t         j                  d«      S )Nr  r#  r    r!   r"   rA   zcache_knobs.<lambda>U  s   € ´u×7KÑ7KÈGÓ7T€ r!   ÚdirÚTRITON_CACHE_MANAGERr   zenv_class[CacheManager]Úmanager_classÚTRITON_REMOTE_CACHE_BACKENDr   zenv_class[RemoteCacheBackend]Úremote_manager_classc                óX   — t         j                  j                  | j                  d|«      S )Nz.triton)r(   rœ   r¸   r  )rC   r¹   s     r"   r$  zcache_knobs.get_triton_dirZ  s   € Ü�w‰w�|‰|˜DŸM™M¨9°gÓ>Ð>r!   N)r¹   r5   r`   r5   )r   r   r   re   r  r¬   r%  r)  r,  r‡   r.  r0  r$  r    r!   r"   r  r  P  sz   … Ù Ñ/OÓP€HˆgÓPáÐ 1Ñ3WÓX€HˆgÓXÙ#Ð$9Ñ;cÓd€L�'ÓdÙÐ-Ñ/TÓU€CˆÓUá-6Ð7MÈ~Ó-^€MÐ*Ó^Ù:CÐDaÐcwÓ:xÐÐ7Ó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<   dZded<   y)Úcompilation_knobsÚTRITON_KERNEL_OVERRIDErj   r(  ÚTRITON_KERNEL_DUMPÚdump_irÚ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_globalsNz Union[CompilationListener, None]Úlistener)r   r   r   rj   r(  r¬   r6  r8  r:  rÆ   r<  r>  r@  rB  rD  rE  r    r!   r"   r3  r3  ^  s    … Ù!Ð":Ó;€HˆhÓ;Ù Ð!5Ó6€GˆXÓ6Ù"*Ð+EÓ"FÐ�xÓFÙ'Ð(?Ó@€N�HÓ@á)¨,Ó7€J�Ó7Ù$Ð%9Ó:€K�Ó:Ù"*Ð+EÓ"FÐ�xÓFÙ$,Ð-IÓ$JÐ˜ÓJÙ,4Ð5YÓ,ZÐ ÓZØ15€HÐ.Ô5r!   r3  c                  óB   — e Zd ZU  ed«      Zded<    ed«      Zded<   y)Úautotuning_knobsÚTRITON_CACHE_AUTOTUNINGrj   r  ÚTRITON_PRINT_AUTOTUNINGÚprintN)r   r   r   rj   r  r¬   rJ  r    r!   r"   rG  rG  l  s"   … ÙÐ8Ó9€Eˆ8Ó9ÙÐ8Ó9€Eˆ8Ô9r!   rG  c                  ó   — e Zd Zdd„Zy)Ú
LaunchHookc                 ó   — y r'   r    )rC   rÛ   s     r"   rß   zLaunchHook.__call__s  s   € Ør!   N)rÛ   r   r`   ra   râ   r    r!   r"   rL  rL  q  s   „ ôr!   rL  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<   y)ÚJITHookCompileInfor5   r+   zdict[KernelParam, str]Ú	signaturer6   Údevicera   Ú	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"   rO  rO  |  sI   … Ø	ƒHØ%Ó%ØƒKØƒOØƒNØƒMØƒOØÓØ!Ó!Ø,Ó,Ø:Ó:ØÓØ„Or!   rO  c                  ó   — e Zd Z	 	 	 	 dd„Zy)ÚJITHookc                ó   — y r'   r    )rC   r+   ÚreprÚfnÚcompileÚis_manual_warmupÚalready_compileds          r"   rß   zJITHook.__call__Ž  rà   r!   N)r+   r5   r_  r5   r`  r   ra  rO  rb  r$   rc  r$   r`   zOptional[bool]râ   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<   d	Z	d
ed<   d	Z
d
ed<   d	Zded<   d	Zded<   y	)Úruntime_knobsÚTRITON_INTERPRETrj   Ú	interpretÚTRITON_DEBUGÚdebugÚTRITON_OVERRIDE_ARCHrÆ   Úoverride_archNzOptional[LaunchHook]Úlaunch_enter_hookÚlaunch_exit_hookzOptional[JITHook]Újit_cache_hookÚjit_post_compile_hook)r   r   r   rj   rg  r¬   ri  rÆ   rk  rl  rm  rn  ro  r    r!   r"   re  re  “  sd   … Ù"Ð#5Ó6€IˆxÓ6Ù˜~Ó.€Eˆ8Ó.Ù!,Ð-CÓ!D€M�;ÓDà.2ÐÐ+Ó2Ø-1ÐÐ*Ó1ð )-€NÐ%Ó,ð 04ÐÐ,Ô3r!   re  c                  óD   — e Zd ZU  ed«      Zded<    edd«      Zded<   y)	Úlanguage_knobsÚTRITON_F32_DEFAULTrÆ   Úfp32_defaultÚTRITON_DEFAULT_FP_FUSIONTrj   Údefault_fp_fusionN)r   r   r   rÆ   rs  r¬   rj   ru  r    r!   r"   rq  rq  ¢  s%   … Ù +Ð,@Ó A€L�+ÓAÙ"*Ð+EÀtÓ"LÐ�xÔLr!   rq  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<   y)Únvidia_knobsÚ	cuobjdumpr¯   ÚnvdisasmÚptxasÚNVPTX_ENABLE_DUMPrj   Ú
dump_nvptxÚDISABLE_PTXAS_OPTÚdisable_ptxas_optÚTRITON_MOCK_PTX_VERSIONrÆ   Úmock_ptx_versionÚTRITON_LIBDEVICE_PATHÚlibdevice_pathÚTRITON_LIBCUDA_PATHÚlibcuda_pathN)r   r   r   r¯   rx  r¬   ry  rz  rj   r|  r~  rÆ   r€  r‚  r„  r    r!   r"   rw  rw  §  s�   … Ù!0°Ó!=€IˆÓ=Ù /°
Ó ;€HˆoÓ;Ù,¨WÓ5€Eˆ?Ó5á#Ð$7Ó8€J�Ó8Ù"*Ð+>Ó"?Ð�xÓ?Ù$/Ð0IÓ$JÐ�kÓJá"-Ð.EÓ"F€N�KÓFÙ +Ð,AÓ B€L�+ÔBr!   rw  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<   y)Ú	amd_knobsÚAMDGCN_USE_BUFFER_OPSrj   Úuse_buffer_opsÚAMDGCN_ENABLE_DUMPÚdump_amdgcnÚTRITON_LIBHIP_PATHrÆ   Úlibhip_pathÚTRITON_HIP_LLD_PATHÚlld_pathÚTRITON_HIP_USE_BLOCK_PINGPONGrÈ   Úuse_block_pingpongÚ"TRITON_HIP_USE_IN_THREAD_TRANSPOSEÚuse_in_thread_transposeÚTRITON_HIP_GLOBAL_PREFETCHr{   Úglobal_prefetchÚTRITON_HIP_LOCAL_PREFETCHÚlocal_prefetchÚTRITON_HIP_USE_ASYNC_COPYÚuse_async_copyÚAMDGCN_SCALARIZE_PACKED_FOPSÚscalarize_packed_fopsN)r   r   r   rj   rˆ  r¬   rŠ  rÆ   rŒ  rŽ  rÈ   r�  r’  r{   r”  r–  r˜  rš  r    r!   r"   r†  r†  ´  s§   … Ù'Ð(?Ó@€N�HÓ@Ù$Ð%9Ó:€K�Ó:Ù*Ð+?Ó@€K�Ó@Ù'Ð(=Ó>€HˆkÓ>ñ (4Ð4SÓ'TÐ˜ÓTÙ,8Ð9]Ó,^Ð˜\Ó^á&Ð'CÓD€O�WÓDÙ%Ð&AÓB€N�GÓBÙ'Ð(CÓD€N�HÓDÙ&.Ð/MÓ&NÐ˜8ÔNr!   r†  c                  ó(   — e Zd ZU  ed«      Zded<   y)Úproton_knobsÚTRITON_CUPTI_LIB_PATHrÆ   Ú	cupti_dirN)r   r   r   rÆ   rž  r¬   r    r!   r"   rœ  rœ  Ä  s   … Ù(Ð)@ÓA€Iˆ{ÔAr!   rœ  )r+   r5   r`   úOptional[str])r+   r5   r/   rŸ  r`   ra   )r7   r   r`   z!Union[None, tuple[Optional[str]]])WÚ
__future__r   r”   r(   r¤   r¡   r¼   Údataclassesr   Ú
contextlibr   Útypingr   r   r   r	   r
   r   r   r   r   r   r   r   Úruntime.cacher   r   Úruntime.jitr   r   Úcompiler.compilerr   r   r   r   rU   r%   r¬   r)   r0   r9   r:   r;   r=   r5   re   r$   rj   r6   r{   rƒ   r…   r‡   r›   r¯   rÆ   rÈ   rË   rØ   rã   rä   rþ   r  r  r  r3  rG  rL  ÚlistÚ
KernelAttrrO  r]  re  rq  rw  r†  rœ  ÚbuildÚredisr  ÚcompilationÚ
autotuningÚruntimeÚlanguager¶   ÚamdÚprotonr    r!   r"   ú<module>r±     sì  ðÞ "ã Û 	Û 	Û Û å !Ý %ß ~× ~× ~Ó ~áß?ß9ß@Ñ@÷	ñ 	ñ 
ƒe€à€ˆtÓ ó3ó
óñ& �)Ó
€Ù
�)Ó
€ô+$ˆw�w Ð'Ñ(ô +$ô\ˆh�s˜C�xÑ ô ô>ˆx˜˜d˜
Ñ#ô >ô	Yˆh�s˜C�xÑ ô 	Yô$�7˜7 GÐ+Ñ,¨h°xÀÑ7HÈ(ÐSZÑJ[Ð7[Ñ.\ô $ñ �KÓ €	ô*�˜	Ñ" L°°i±À$ÀyÁ/Ð1QÑ$Rô *ð& ÷ð ó ðô$�h˜s J˜Ñ/ô ôB	�,˜s C˜xÑ(¨'ô 	ô	�<  d 
Ñ+¨Xô 	ñ �$Ô÷Qð Qó ðQô4˜(ô ñ �\¨Ô6€
÷'&ñ '&ôT�ô ô\�*ô \ô7�*ô 7ð Ó ô?�*ô ?ô6˜
ô 6ô:�zô :ô
�ô ð �%˜˜S˜‘/Ñ"€
ô˜ô ô ˆhô ô4�Jô 4ôM�Zô Mô

C�:ô 
CôO�
ô Oô B�:ô Bñ 	‹€Ù‹€Ù‹€ÙÓ!€ÙÓ€
Ù
‹/€ÙÓ€Ù	‹€Ùƒk€Ù	‹�r!   