+
    ‹°³j¢R  ã                  óÎ  € ^ RI Ht ^ RIt^ RIt^RIHtHt ^RIHt ^RIH	t	 ^RIH
t
Ht ^RIHtHt ^RIHt ^R	IHtHtHtHt ^R
IHt ^RIHt ^ RIHt ^ RIt^ RIt^ RIt^ RIt^ RI t Rt!R]!/t"Rt#R]#/t$R t% ! R R4      t& ! R R4      t']PP                  ! 4       R 4       t)R t*R R lt+ ! R R4      t,R%R lt-R R lt. ! R R4      t/ ! R  R!]04      t1R" t2 ! R# R$4      t3R# )&é    )ÚannotationsN)Úget_cache_invalidating_env_varsÚir)Úbackends)ÚLanguage)ÚBaseBackendÚ	GPUTarget)Ú__version__Úknobs)ÚOutOfResources)Úget_cache_managerÚget_dump_managerÚget_override_managerÚget_cache_key)Údriver)Úget_sass)ÚPathz=\.(?:visible|extern)\s+\.(?:entry|func)\s+(\w+)\s*\(([^)]*)\)Úptxz\.param\s+\.(\w+)c                óè   € \         P                  ! R V 4      p\         P                  ! RV 4      pVe   R# \         P                  ! RRV 4      p Ve"   R\        VP	                  ^4      4      ,           # V # )z!tt\.ptr<([^,]+)ztt.nv_tma_desc = 1Ú	nvTmaDescz {[^}]+}Ú Ú*)ÚreÚsearchÚsubÚconvert_type_reprÚgroup)ÚxÚmatchÚtmas   &  ÚY/home/luca/screen-checking/.venv/lib/python3.14/site-packages/triton/compiler/compiler.pyr   r   '   sd   € ô �IŠIÐ)¨1Ó-€EÜ
�)Š)Ð)¨1Ó
-€CØ
‚ÙÜ
�Šˆ{˜B Ó"€AØÒØÔ& u§{¡{°1£~Ó6Õ6Ð6Ø€Hó    c                  ó>   € ] tR t^4tR	R R lltR tR R ltR tRtR# )
Ú	ASTSourceNc               ó   € V ^8„  d   QhRR/# ©é   ÚreturnÚNone© )Úformats   "r!   Ú__annotate__ÚASTSource.__annotate__6   s   € ÷ Añ AÀdñ Ar"   c                	ó:  € Wn         \        P                  V n        R V n        VP
                  V n        W n        \        4       V n	        Vet   VP                  4        F_  w  rV\        V\        4      '       d   VP                  P                  V4      3MTp\        V\        4      '       g   Q hW`P                  V&   Ka  	  T;'       g    \        4       V n        V P                  P#                  4        F$  p\        V\        4      '       d   K  \%        R4      h	  R# )ÚttirNzSignature keys must be string)Úfnr   ÚTRITONÚlanguageÚextÚ__name__ÚnameÚ	signatureÚdictÚ	constantsÚitemsÚ
isinstanceÚstrÚ	arg_namesÚindexÚtupleÚattrsÚkeysÚ	TypeError)Úselfr0   r6   Ú
constexprsr?   ÚkÚvs   &&&&&  r!   Ú__init__ÚASTSource.__init__6   sÒ   € ØŒÜ Ÿ™ˆŒØˆŒØ—K‘KˆŒ	Ø"ŒÜ›ˆŒØÒ!Ø"×(Ñ(Ö*‘�Ü1;¸A¼s×1CÒ1C�R—\‘\×'Ñ'¨Ó*Ñ-È�Ü! !¤U×+Ò+Ð+Ð+Ø$%—‘˜qÓ!ñ +ð —_�_œd›fˆŒ
Ø—‘×$Ñ$Ö&ˆAÜ˜a¤×%Ô%ÜÐ ?Ó@Ð@ó 'r"   c           	     	óò  € \        V P                  P                  4       4       UUu. uF  w  rVNK	  	  pppR  pRP                  \        V P                  P                  4       4       UUu. uF  w  rV! V4      NK  	  upp4      pV P
                  P                   R\        V P                  4       RV RV 2p\        P                  ! VP                  R4      4      P                  4       # u uppi u uppi )c                óT   € \        V R 4      '       d   V P                  # \        V 4      # )Ú	cache_key)ÚhasattrrJ   r;   )r   s   &r!   Ú<lambda>Ú ASTSource.hash.<locals>.<lambda>I   s    € ¬7°1°k×+BÒ+B˜AŸK™KÐNÌÈAËÐNr"   Ú-úutf-8)Úsortedr6   r9   Újoinr8   r0   rJ   r;   r?   ÚhashlibÚsha256ÚencodeÚ	hexdigest)rB   rD   rE   Ú
sorted_sigÚget_keyÚconstants_keyÚkeys   &      r!   ÚhashÚASTSource.hashG   sÂ   € Ü$*¨4¯>©>×+?Ñ+?Ó+AÔ$BÔCÑ$B™D˜A“aÑ$Bˆ
ÑCÙNˆØŸ™¼ÀÇÁ×@TÑ@TÓ@VÔ9WÔ!XÑ9W±°¡'¨!¦*Ñ9WÒ!XÓYˆØ—‘×"Ñ"Ð# 1¤S¨¯©£_Ð$5°Q°z°lÀ!ÀMÀ?ÐSˆÜ�~Š~˜cŸj™j¨Ó1Ó2×<Ñ<Ó>Ð>ùó	 Dùã!Xs   §C-Á+C3
c               ó   € V ^8„  d   QhRR/# ©r'   Útargetr	   r*   )r+   s   "r!   r,   r-   N   s   € ÷ 2ñ 2˜iñ 2r"   c           	     	ó:   € ^RI Hp V! V P                  WW#VR7      # )é   )Úast_to_ttir)ÚcontextÚoptionsÚcodegen_fnsÚ
module_map)Úcode_generatorra   r0   )rB   r^   rc   rd   re   rb   ra   s   &&&&&& r!   Úmake_irÚASTSource.make_irN   s   € Ý/Ù˜4Ÿ7™7 DÀ7Ø&0ô2ð 	2r"   c                	ó   € \        4       # ©N)r7   ©rB   s   &r!   Úparse_optionsÚASTSource.parse_optionsS   s	   € Ü‹vˆr"   )r?   r8   r3   r0   r2   r5   r6   )NN©	r4   Ú
__module__Ú__qualname__Ú__firstlineno__rF   rZ   rg   rl   Ú__static_attributes__r*   r"   r!   r$   r$   4   s   † ÷Aò"?õ2ö
r"   r$   c                  ó4   € ] tR t^WtR tR tR R ltR tRtR# )ÚIRSourcec                	ó  € Wn         \        V4      pVP                  R ,          V n        \        P
                  V n        VP                  4       V n        \        P                  ! V4       VP                  V4       V P                  R8X  dÉ   \        P                  ! \        V P                  ,          V P                  \        P                  4      pVP                  ^4      V n        VP                  ^4      p\        P"                  ! \$        V P                  ,          V4      p\'        V4       UUu/ uF  w  rxV\)        V4      bK  	  uppV n        R# \        P,                  ! V P                   V4      V n        V P.                  P1                  4       p	RV	,           V n        V P.                  P3                  V	4      p
V P.                  P5                  V
4      p\'        V4       UUu/ uF  w  rxWxbK	  	  uppV n        R# u uppi u uppi )ºr`   NNr   Ú@N)Úpathr   Úsuffixr3   r   r1   r2   Ú	read_textÚsrcr   Úload_dialectsr   r   Úprototype_patternÚ	MULTILINEr   r5   ÚfindallÚarg_type_patternÚ	enumerater   r6   Úparse_mlir_moduleÚmoduleÚget_entry_func_nameÚget_functionÚget_function_signature)rB   rx   rb   Úbackendr   r6   ÚtypesrD   ÚtyÚfn_nameÚfuncOpÚfunc_tys   &&&&        r!   rF   ÚIRSource.__init__Y   sp  € ØŒ	Ü�D‹zˆØ—;‘;˜r•?ˆŒÜ Ÿ™ˆŒØ—>‘>Ó#ˆŒÜ
×Ò˜Ô!Ø×Ñ˜gÔ&ð �8‰8�uÔÜ—I’IÔ/°·±Õ9¸4¿8¹8ÄRÇ\Á\ÓRˆEØŸ™ A›ˆDŒIØŸ™ A›ˆIÜ—J’JÔ/°·±Õ9¸9ÓEˆEÜDMÈeÔDTÔUÑDT¹5¸1˜aÔ!2°2Ó!6Ò6ÑDTÒUˆDŽNä×.Ò.¨t¯y©y¸'ÓBˆDŒKØ—k‘k×5Ñ5Ó7ˆGØ˜g�ˆDŒIØ—[‘[×-Ñ-¨gÓ6ˆFØ—k‘k×8Ñ8¸Ó@ˆGÜ1:¸7Ô1CÔDÑ1C©¨˜ašeÑ1CÒDˆDŽNùó Vùó Es   Ä2G>Ç&Hc                	ó|   € \         P                  ! V P                  P                  R 4      4      P	                  4       # )rO   )rR   rS   r{   rT   rU   rk   s   &r!   rZ   ÚIRSource.hashr   s'   € Ü�~Š~˜dŸh™hŸo™o¨gÓ6Ó7×AÑAÓCÐCr"   c               ó   € V ^8„  d   QhRR/# r]   r*   )r+   s   "r!   r,   ÚIRSource.__annotate__u   s   € ÷ ñ ˜iñ r"   c                	ó:   € WPP                   n        V P                   # rj   )rƒ   rb   )rB   r^   rc   rd   re   rb   s   &&&&&&r!   rg   ÚIRSource.make_iru   s   € Ø%�‰ÔØ�{‰{Ðr"   c                	óÖ   € V P                   R 8X  dP   V P                  P                  R4      pVf   Q R4       hRV/pV P                  P                  R4      pVe   W2R&   V# \        4       # )Úttgirzttg.num-warpsz'Unable to parse ttg.num-warps attributeÚ	num_warpszttg.num-ctasÚnum_ctas)r3   rƒ   Úget_int_attrr7   )rB   r–   rc   r—   s   &   r!   rl   ÚIRSource.parse_optionsy   sl   € Ø�8‰8�wÔØŸ™×0Ñ0°ÓAˆIØÒ(ÐSÐ*SÓSÐ(Ø" IÐ.ˆGØ—{‘{×/Ñ/°Ó?ˆHØÒ#Ø&.˜
Ñ#ØˆNÜ‹vˆr"   )r3   r2   rƒ   r5   rx   r6   r{   Nrn   r*   r"   r!   rt   rt   W   s   † òEò2Dõö	r"   rt   c                ób   € \         P                  P                  P                  V 4      R ,          # )Úmax_shared_mem)r   ÚactiveÚutilsÚget_device_properties)Údevices   &r!   r›   r›   …   s#   € ä�=‰=×Ñ×4Ñ4°VÓ<Ð=MÕNÐNr"   c                ó  € VR 8X  g   VR8X  d   \         P                  ! W4      pW#n        V# VR8X  g   VR8X  g   VR8X  d   \        V 4      P	                  4       # VR8X  g   VR8X  d   \        V 4      P                  4       # R# )r/   r•   Úllirr   ÚamdgcnÚcubinÚhsacoN)r   r‚   rb   r   rz   Ú
read_bytes)Ú	full_namer3   rb   rƒ   s   &&& r!   Úparser§   Š   st   € Ø
ˆf„}˜˜wœÜ×%Ò% iÓ9ˆØ ŒØˆØ
ˆf„}˜˜uœ¨¨x¬Ü�I‹×(Ñ(Ó*Ð*Ø
ˆg„~˜ œÜ�I‹×)Ñ)Ó+Ð+ñ (r"   c               ó   € V ^8„  d   QhRR/# )r'   ÚeÚBaseExceptionr*   )r+   s   "r!   r,   r,   •   s   € ÷ #$ñ #$˜ñ #$r"   c                óÄ  a€ \         P                  P                  '       d   R# V P                  e   \	        V P                  4       V P
                  e   \	        V P
                  4       RR.pV Uu. uF"  q"P                  R\        P                  4      NK$  	  ppV P                  o. pSe`   \        ;QJ d    V3R lV 4       F  '       g   K   RM	  RM! V3R lV 4       4      '       g   VP                  S4       SP                  oKc  \        W3R,          4       F  w  rEWTn        K  	  V'       g
   RV n	        R# RVR	,          n        V^ ,          V n	        R# u upi )
z€
Removes code_generator.py and related files from tracebacks.

These are uninteresting to the user -- "just show me *my* code!"
Nz"/triton/compiler/code_generator.pyz/ast.pyÚ/c              3  ó–   <"  € T F>  pSP                   P                  P                  P                  V4      '       g   K:  Vx € K@  	  R # 5irj   )Útb_frameÚf_codeÚco_filenameÚendswith)Ú.0ÚfÚtbs   & €r!   Ú	<genexpr>Ú#filter_traceback.<locals>.<genexpr>­   s1   øé € ÐV™i˜¨2¯;©;×+=Ñ+=×+IÑ+I×+RÑ+RÐST×+U—1’1›iùs
   ƒ7A	¿
A	TFrv   éÿÿÿÿ)r   ÚcompilationÚfront_end_debuggingÚ	__cause__Úfilter_tracebackÚ__context__ÚreplaceÚosÚsepÚ__traceback__ÚanyÚappendÚtb_nextÚzip)r©   Ú	BAD_FILESÚbad_fileÚframesÚ	cur_frameÚ
next_framer´   s   &     @r!   r»   r»   •   s
  ø€ ô ×Ñ×,×,Ð,Ùà‡{�{ÒÜ˜Ÿ™Ô%Ø‡}�}Ò Ü˜Ÿ™Ô'ð 	-Øð€Iñ @IÓI¹y°8×!Ñ! #¤r§v¡vÖ.¹y€IÐIà	
�‰€BØ€FØ
Š.ß‹sÔV™iÓV�s�sŠsÔV™iÓV×VÒVØ�M‰M˜"ÔØ�Z‰ZŠä#& v°b­zÖ#:ÑˆØ&Öñ $;÷ ØˆŽà!ˆˆr�
ÔØ  �)ˆŽùò! Js   Á2(Ec                  óF   € ] tR t^»tR R ltR R ltR R ltR R ltR	tR
# )ÚCompileTimerc               ó   € V ^8„  d   QhRR/# r&   r*   )r+   s   "r!   r,   ÚCompileTimer.__annotate__½   s   € ÷ 4ñ 4˜$ñ 4r"   c                	ód   € \         P                   ! 4       V n        R V n        . V n        R V n        R # rj   )ÚtimeÚstartÚir_initialization_endÚlowering_stage_endsÚstore_results_endrk   s   &r!   rF   ÚCompileTimer.__init__½   s'   € Ü ŸIšI›KˆŒ
Ø37ˆÔ"Ø<>ˆÔ Ø/3ˆÖr"   c               ó   € V ^8„  d   QhRR/# r&   r*   )r+   s   "r!   r,   rÍ   Ã   s   € ÷ 1ñ 1¨Dñ 1r"   c                	ó:   € \         P                   ! 4       V n        R # rj   )rÏ   rÑ   rk   s   &r!   Úfinished_ir_initializationÚ'CompileTimer.finished_ir_initializationÃ   s   € Ü%)§Y¢Y£[ˆÖ"r"   c               ó    € V ^8„  d   QhRRRR/# )r'   Ú
stage_namer;   r(   r)   r*   )r+   s   "r!   r,   rÍ   Æ   s   € ÷ Cñ C¨ð C°ñ Cr"   c                	óf   € V P                   P                  V\        P                  ! 4       34       R # rj   )rÒ   rÂ   rÏ   )rB   rÚ   s   &&r!   Ústage_finishedÚCompileTimer.stage_finishedÆ   s!   € Ø× Ñ ×'Ñ'¨´T·Y²Y³[Ð(AÖBr"   c               ó   € V ^8„  d   QhRR/# )r'   r(   zknobs.CompileTimesr*   )r+   s   "r!   r,   rÍ   É   s   € ÷ 
ñ 
Ð'ñ 
r"   c                	óp  € \         P                   ! 4       pV P                  f   Wn        MWn        R R lp. pV P                  pV P                   F  w  rVVP	                  WR! WF4      34       TpK!  	  \
        P                  ! V! V P                  V P                  4      VV! W@P                  4      R7      # )Nc               ó$   € V ^8„  d   QhRRRRRR/# )r'   rÐ   ÚfloatÚendzfloat | Noner(   Úintr*   )r+   s   "r!   r,   Ú&CompileTimer.end.<locals>.__annotate__Ð   s!   € ÷ 	0ñ 	0˜ð 	0 \ð 	0°cñ 	0r"   c                ó>   € Vf   ^ # \        W,
          R,          4      # )Ni@B )rã   )rÐ   râ   s   &&r!   ÚdeltaÚCompileTimer.end.<locals>.deltaÐ   s   € ØŠ{ÙÜ˜� wÕ.Ó/Ð/r"   )Úir_initializationÚlowering_stagesÚstore_results)rÏ   rÑ   rÓ   rÒ   rÂ   r   ÚCompileTimesrÐ   )rB   Ú	timestampræ   Úlowering_stage_durationsÚstage_startrÚ   Ú	stage_ends   &      r!   râ   ÚCompileTimer.endÉ   s§   € Ü—I’I“Kˆ	Ø×%Ñ%Ò-Ø)2Õ&à%.Ô"õ	0ð
 $&Ð Ø×0Ñ0ˆØ%)×%=Ô%=Ñ!ˆJØ$×+Ñ+¨Z¸¸{Ó9VÐ,WÔXØ#ŠKñ &>ô ×!Ò!Ù# D§J¡J°×0JÑ0JÓKØ4Ù ×-CÑ-CÓDô
ð 	
r"   )rÑ   rÒ   rÐ   rÓ   N)	r4   ro   rp   rq   rF   r×   rÜ   râ   rr   r*   r"   r!   rË   rË   »   s   † õ4õ1õC÷
ñ 
r"   rË   c                ó€  € \         P                  P                  pV'       d   \        4       pVf   \        P
                  P                  4       p\        V\        4      '       g   Q R4       h\        V4      p\        V \        4      '       * pV'       d?   \        V \        4      '       g   Q R4       h\        P                  ! 4       p\        WV4      p V P                  4       p	TP                  \!        T;'       g    \!        4       3/ V	B 4      pVf   \#        4       MTp
\%        WW*R7      p\         P&                  P(                  e)   \         P&                  P)                  4       w  rÍW¼,          p\*        P,                  ! VP/                  R4      4      P1                  4       p\3        V4      p\         P                  P4                  p\         P                  P6                  p\         P                  P8                  pV'       d   \;        V P=                  4       4      MR pV'       d   \?        V P=                  4       4      MR pV P@                  R,          pV R2pVPC                  V4      ;'       g    / pVPE                  V4      p\         P                  PF                  pV'       gO   VeK   \I        V VV4      pV'       d4   V! V VPJ                  PM                  4       VXPO                  4       RR7       V# R	VR
V/VPP                  CV
Cp\R        VR&   \!        4       pVPU                  VW PV                  4       \Y        VP[                  4       4      P]                  V P^                  4      pV'       d
   V^,          p\        V \        4      '       g=   \        P                  ! 4       p\        P`                  ! V4       VPa                  V4       VPc                  V4      pVPe                  4       p V Pg                  WVVX4      p T'       d(   T RT P^                   2p"TPm                  T T"4      TT"&   MT R2p"TPm                  T T"4      TT"&   \         P                  Pn                  p#T'       d<   T#'       d4   T Pq                  T Pr                  4       \u        RT Pr                   24       T'       d   XPw                  4        \Y        TPy                  4       4      TR   EFK  w  p$p%T%! T T4      p&T RT$ 2p"TfC   TPE                  RR 4      ;p''       d(   T'P{                  RT$ 24      '       d   \}        T'T$T4      p&M4TP                  T"4      ;p('       d   \u        RT( 24       \}        T(T$T4      p&T'       d   T$R9   d   TPm                  T&T"4      TT"&   Te>   TPm                  T&T"4       T$R8X  d%   \�        T&4      p)TPm                  T)TR,           4       T#T$8X  d1   TP                  T"4      p*T&Pq                  T*4       \u        RT* 24       T&p T'       g   EK:  XPƒ                  T$4       EKN  	  TPm                  \„        P†                  ! T\ˆ        R7      TRR7      TT&   TP‹                  TT4       T'       d   T! T TTXPO                  4       RR7       \I        T TT4      #   \h         d   p!\k        T!4       h R p!?!ii ; i)Nz target must be of GPUTarget typez'source must be either AST or a filepath)Úenv_varsrO   :Né–   Nú.jsonT)r{   ÚmetadataÚmetadata_groupÚtimesÚ	cache_hitrZ   r^   Útriton_versionÚ.z.sourcezCreating new locations for Úir_overridez
Overriding kernel with file r£   z.sass)ÚdefaultF)Úbinary)r£   r¤   Újson)Fr   r¸   ÚlistenerrË   r   rœ   Úget_current_targetr:   r	   Úmake_backendr$   r;   r   rb   rt   rl   r7   r   r   ÚruntimeÚadd_stages_inspection_hookrR   rS   rT   rU   r   ÚoverrideÚdump_irÚstore_binary_onlyr   rZ   r   r5   Ú	get_groupÚgetÚalways_compileÚCompiledKernelrõ   Ú_asdictrâ   Ú__dict__r
   Ú
add_stagesr2   Úlistr@   r=   r3   r|   Úget_codegen_implementationÚget_module_maprg   Ú	Exceptionr»   ÚputÚ
use_ir_locÚcreate_location_snapshotrx   Úprintr×   r9   r±   r§   Úget_filer   rÜ   rþ   ÚdumpsÚvarsÚ	put_group)+r{   r^   rc   Ú	_env_varsÚcompilation_listenerÚtimerr‡   Ú	ir_sourcerb   Úextra_optionsrò   rY   Úinspect_stages_keyÚinspect_stages_hashrZ   Úfn_cache_managerÚenable_overrideÚenable_ir_dumpÚstore_only_binaryÚfn_override_managerÚfn_dump_managerÚ	file_nameÚmetadata_filenamerö   Úmetadata_pathr	  Úresrõ   ÚstagesÚfirst_stagerd   re   rƒ   r©   Úir_filenamer  r3   Ú
compile_irÚnext_modulerû   r¦   ÚsassÚir_full_names+   &&&&                                       r!   Úcompiler2  â   sÈ  € Ü ×,Ñ,×5Ñ5ÐßÜ“ˆà‚~Ü—‘×1Ñ1Ó3ˆÜ�fœi×(Ò(ÐLÐ*LÓLÐ(Ü˜6Ó"€GÜ˜s¤IÓ.Ô.€IçÜ˜#œs×#Ò#ÐNÐ%NÓNÐ#Ü—*’*“,ˆÜ�s WÓ-ˆà×%Ñ%Ó'€MØ×#Ñ#¤D¨×):Ð):´D³FÑ$L¸mÑ$LÓM€Gà4=Ò4EÔ.Ô0È9€HÜ
˜ gÔ
A€CÜ‡}�}×/Ñ/Ò;Ü27·-±-×2ZÑ2ZÓ2\Ñ/ÐØÕ!ˆÜ�>Š>˜#Ÿ*™* WÓ-Ó.×8Ñ8Ó:€DÜ(¨Ó.Ðô ×'Ñ'×0Ñ0€OÜ×&Ñ&×.Ñ.€NÜ×)Ñ)×;Ñ;Ðß>MÔ.¨s¯x©x«zÔ:ÐSWÐß6DÔ& s§x¡x£zÔ2È$€Oð
 —‘˜•€IØ$˜+ UÐ+ÐØ%×/Ñ/Ð0AÓB×HÐHÀb€NØ"×&Ñ&Ð'8Ó9€MÜ×&Ñ&×5Ñ5€Nß˜mÒ7ä˜S .°$Ó7ˆßÙ ØØŸ™×-Ñ-Ó/Ø-Ø—i‘i“kØõð ˆ
ð 	�Ø�&ðð ×
Ñ
ðð ð	€Hô "-€HÐÑä‹V€FØ×Ñ�v˜w¯©Ô5Ü�v—{‘{“}Ó%×+Ñ+¨C¯G©GÓ4€KçØ�qÕˆô �cœ8×$Ò$Ü—*’*“,ˆÜ
×Ò˜Ô!Ø×Ñ˜gÔ&à×4Ñ4°WÓ=€KØ×'Ñ'Ó)€JðØ—‘˜V¨k¸:ÀwÓOˆ÷
 Ø"˜ 1 S§W¡W IÐ.ˆØ&6×&:Ñ&:¸6À;Ó&Oˆ�{Ò#à"˜ 7Ð+ˆØ&6×&:Ñ&:¸6À;Ó&Oˆ�{Ñ#ä×"Ñ"×-Ñ-€Jß—ZØ×'Ñ'¨¯©Ô1ÜÐ+¨C¯H©H¨:Ð6Ô7çØ×(Ñ(Ô*Ü §¡£Ó/°°Ô=‰ˆˆZÙ  ¨Ó2ˆØ"˜ 1 S EÐ*ˆØÒ&ð  (Ÿ|™|¨M¸4Ó@Ð@�Ö@Àk×FZÑFZÐ]^Ð_bÐ^cÐ[d×FeÒFeÜ# K°°gÓ>�øØ-×6Ñ6°{ÓCÐCˆYÖCÜÐ2°9°+Ð>Ô?Ü 	¨3°Ó8ˆKç! sÐ.HÔ'HØ*:×*>Ñ*>¸{ÈKÓ*XˆN˜;Ñ'ØÒ&Ø×Ñ ¨[Ô9Ø�gŒ~Ü Ó,�Ø×#Ñ# D¨)°gÕ*=Ô>à˜ÔØ+×4Ñ4°[ÓAˆLØ×0Ñ0°Ô>ÜÐ/°¨~Ð>Ô?ØˆßÒØ× Ñ  ×%ñ5 >ð8 )9×(<Ñ(<¼T¿ZºZÈÔZ^Ô=_ÐarØDIð )=ó )K€NÐ$Ñ%à×ÑÐ0°.ÔA÷ Ù ¨xÈÐ^c×^gÑ^gÓ^iØ',õ	.ô ˜#˜~¨tÓ4Ð4øôo ô Ü˜ÔØûðús   Ï0Z! Ú!Z=Ú,Z8Ú8Z=c               ó    € V ^8„  d   QhRRRR/# )r'   r^   r	   r(   r   r*   )r+   s   "r!   r,   r,   n  s   € ÷ ñ ˜ð  {ñ r"   c                ó8  € \         P                  ! 4        Uu. uF1  qP                  P                  V 4      '       g   K%  VP                  NK3  	  pp\	        V4      ^8w  d(   \        \	        V4       RV P                   RV R24      hV^ ,          ! V 4      # u upi )r`   z! compatible backends for target (z) (z). There should only be one.)r   ÚvaluesÚcompilerÚsupports_targetÚlenÚRuntimeErrorr‡   )r^   r   Úactivess   &  r!   r  r  n  sˆ   € Ü#+§?¢?Ô#4Ó[Ñ#4˜a¿
¹
×8RÑ8RÐSY×8ZŒzˆq�zŒzÑ#4€GÐ[Ü
ˆ7ƒ|�qÔÜÜ�7‹|ˆnÐ=¸f¿n¹nÐ=MÈSÐQXÐPYÐYuÐvóxð 	xà�1Ž:�fÓÐùò	 \s
   ˜"B¿Bc                  ó(   € ] tR tRtR tR tR tRtR# )ÚLazyDictiv  c                	ó    € Wn         . V n        R # rj   ©ÚdataÚextras)rB   r?  s   &&r!   rF   ÚLazyDict.__init__x  s   € ØŒ	ØˆŽr"   c                	ó°   € V P                    F!  w  rV P                  V! V!  ,          V n        K#  	  V P                   P                  4        V P                  # rj   )r@  r?  Úclear©rB   ÚfuncÚargss   &  r!   r  ÚLazyDict.get|  s@   € ØŸ+œ+‰JˆDØŸ	™	¡D¨$¡KÕ/ˆDŽIñ &à�‰×ÑÔØ�y‰yÐr"   c                	ó>   € V P                   P                  W34       R # rj   )r@  rÂ   rD  s   &&&r!   ÚaddÚLazyDict.add‚  s   € Ø�‰×Ñ˜D˜<Ö(r"   r>  N)r4   ro   rp   rq   rF   r  rI  rr   r*   r"   r!   r<  r<  v  s   † òòö)r"   r<  c                  ó   € ] tR tRtR tRtR# )ÚAsmDicti†  c                	óf   € VR 8X  d   \        V R,          4      pM\        RV,          4      hW V&   V# )r0  r£   zUnknown key: '%s')r   ÚKeyError)rB   rY   Úvalues   && r!   Ú__missing__ÚAsmDict.__missing__ˆ  s4   € à�&Œ=Ü˜T '�]Ó+‰EäÐ.°Õ4Ó5Ð5àˆS‰	Øˆr"   r*   N)r4   ro   rp   rq   rP  rr   r*   r"   r!   rL  rL  †  s   † ör"   rL  c                ó.   € \         P                  ! V 4      hrj   )ÚcopyÚdeepcopy)ÚerrrF  Úkwargss   &*,r!   Ú_raise_errorrW  “  s   € Ü
�-Š-˜Ó
Ðr"   c                  óD   € ] tR tRtR tR tR t]R 4       tR t	R t
RtR	# )
r
  i—  c           	     	ó  € ^ RI Hp \        R VP                  4        4       4      p\        P
                  ! VP                  4       4      pVR,          p\        VR,          VR,          VR,          4      VR&   V! R\        \        VP                  4       4      4      4      pV! R/ VB V n        \        V P                  P                  4      p	V	P                  V P                  4      V n        Wn        W0n        V P                  P$                  V n        VP                  4        U
Uu. uF)  w  r«V
P'                  R4      '       d   K  \)        V4      NK+  	  pp
pV	P*                  p\-        V Uu/ uFM  pVP.                  R	,          VP.                  R	,          V8X  d   VP1                  4       MVP                  4       bKO  	  up4      V n        W n        V P2                  V,          V n        R
V n        R
V n        R
V n        R
# u upp
i u upi )r   )Ú
namedtuplec              3  ón   "  € T F+  w  rVP                  R 4      '       g   K  \        V4      x € K-  	  R# 5i)rô   N)r±   r   )r²   ÚcÚps   &  r!   rµ   Ú*CompiledKernel.__init__.<locals>.<genexpr>›  s(   é € Ð`Ñ2H©$¨!ÈAÏJÉJÐW^×L_œgœd 1Ÿg˜gÓ2Hùs   ‚5¢5r^   r‡   ÚarchÚ	warp_sizeÚKernelMetadatarô   rv   Nr*   )ÚcollectionsrZ  Únextr9   rþ   Úloadsrz   r	   rP   r  r@   rõ   r  r^   Úpack_metadataÚpacked_metadatar{   rZ   r5   r±   r   Ú
binary_extrL  ry   r¥   Úasmrö   Úkernelrƒ   ÚfunctionÚ_run)rB   r{   rö   rZ   rZ  r)  rõ   r^   ra  r‡   r\  r]  Ú	asm_filesrg  Úfiles   &&&&           r!   rF   ÚCompiledKernel.__init__™  s¡  € Ý*ÜÑ`°.×2FÑ2FÔ2HÓ`ÓaˆÜ—:’:˜m×5Ñ5Ó7Ó8ˆà˜(Õ#ˆÜ& v¨iÕ'8¸&À½.È&ÐQ\ÕJ]Ó^ˆ�ÑÙ#Ð$4´f¼TÀ(Ç-Á-Ã/Ó=RÓ6SÓTˆÙ&Ñ2¨Ñ2ˆŒÜ˜tŸ}™}×3Ñ3Ó4ˆØ&×4Ñ4°T·]±]ÓCˆÔØŒØŒ	Ø—M‘M×&Ñ&ˆŒ	à)7×)=Ñ)=Ô)?Ô[Ñ)?¡ ÀqÇzÁzÐRY×GZ”W”T˜!–WÑ)?ˆ	Ñ[Ø×'Ñ'ˆ
Üá!ó
á!�ð �K‰K˜�O°$·+±+¸bµ/ÀZÔ2O˜TŸ_™_Ô.ÐUY×UcÑUcÓUeÒeÙ!ñ
ó ˆŒð -ÔØ—h‘h˜zÕ*ˆŒð ˆŒØˆŒØˆŽ	ùó \ùò
s   Ä(HÅHÅ/AH	c                	ó|  € V P                   e®   \        P                  P                  eV   \        P                  P                  V P                   V P                  V P
                  V P                  V P                  4       \        P                  P                  P                  V P                   4       R V n         R # R # rj   )rƒ   r   r  Úkernel_unload_hookrj  r5   rö   rZ   r   rœ   r�   Úunload_modulerk   s   &r!   Ú__del__ÚCompiledKernel.__del__·  sx   € à�;‰;Ò"Ü�}‰}×/Ñ/Ò;Ü—‘×0Ñ0°·±¸d¿m¹mÈTÏYÉYÐX\×XkÑXkÐmq×mvÑmvÔwä�M‰M×Ñ×-Ñ-¨d¯k©kÔ:ØˆDŽKñ #r"   c                	ó€  a € S P                   e   R # V 3R lp\        P                  P                  4       p\        P                  P	                  S P
                  S P                  4      S n        \        V4      pS P                  P                  V8”  d(   V! \        S P                  P                  VR4      4       \        S P                  R4      '       d„   S P                  P                  el   RpS P                  P                  P                  ^k8X  d   RpS P                  P                  V8”  d(   V! \        S P                  P                  VR4      4       \        P                   P"                  eV   \        P                   P#                  S P                   S P$                  S P&                  S P(                  S P*                  4       \        P                  P,                  P/                  S P&                  S P0                  S P                  P                  V4      w  S n         S n        S n        S n        S n        \        P                  P9                  4       P:                  pS P                  P<                  V,          S P6                  8”  d9   V! \        S P                  P<                  V,          S P6                  R4      4       \        P                   P>                  eX   \        P                   P?                  S P                   S P$                  S P&                  S P(                  S P*                  4       R # R # )Nc                ót   <€ \         P                  ! V 4      p\        P                  ! \        V4      Sn        V hrj   )rS  rT  Ú	functoolsÚpartialrW  rk  )rU  Ú
cloned_errrB   s   & €r!   Úraise_Ú,CompiledKernel._init_handles.<locals>.raise_Ä  s,   ø€ ô Ÿš sÓ+ˆJÜ!×)Ò)¬,¸
ÓCˆDŒIØˆIr"   zshared memoryÚ	tmem_sizei   i@  ztensor memoryÚthreads) rƒ   r   rœ   Úget_current_deviceÚlauncher_clsr{   rõ   rk  r›   Úsharedr   rK   r{  r^   r_  r   r  Úkernel_load_start_hookrj  r5   rö   rZ   r�   Úload_binaryri  Ún_regsÚn_spillsÚn_max_threadsr   r`  r–   Úkernel_load_end_hook)rB   ry  rŸ   Ú
max_sharedÚmax_tmem_sizer`  s   f     r!   Ú_init_handlesÚCompiledKernel._init_handlesÀ  s'  ø€ Ø�;‰;Ò"Ùõ	ô —‘×1Ñ1Ó3ˆä—M‘M×.Ñ.¨t¯x©x¸¿¹ÓGˆŒ	ä# FÓ+ˆ
Ø�=‰=×Ñ *Ô,Ù”> $§-¡-×"6Ñ"6¸
ÀOÓTÔUÜ�4—=‘= +×.Ò.°4·=±=×3JÑ3JÒ3VàˆMØ�}‰}×#Ñ#×(Ñ(¨CÔ/Ø #�Ø�}‰}×&Ñ&¨Ô6Ù”~ d§m¡m×&=Ñ&=¸}ÈoÓ^Ô_Ü�=‰=×/Ñ/Ò;Ü�M‰M×0Ñ0°·±¸d¿m¹mÈTÏYÉYÐX\×XkÑXkÐmq×mvÑmvÔwäU[×UbÑUb×UhÑUh×UtÑUtØ�I‰I�t—{‘{ D§M¡M×$8Ñ$8¸&óVBÑRˆŒ�T”] D¤K°´ÀÔ@Rä—M‘M×4Ñ4Ó6×@Ñ@ˆ	Ø�=‰=×"Ñ" YÕ.°×1CÑ1CÔCÙ”> $§-¡-×"9Ñ"9¸IÕ"EÀt×GYÑGYÐ[dÓeÔfÜ�=‰=×-Ñ-Ò9Ü�M‰M×.Ñ.¨t¯{©{¸D¿M¹MÈ4Ï9É9ÐVZ×ViÑViÐko×ktÑktÖuñ :r"   c                	óV   € V P                   f   V P                  4        V P                   # rj   )rk  rˆ  rk   s   &r!   ÚrunÚCompiledKernel.runç  s"   € à�9‰9ÒØ×ÑÔ Ø�y‰yÐr"   c                	ó2  € \         P                  P                  f   R # V P                  4        \	        RV P
                  RV P                  RV/4      p\        V P                  \        4      '       d#   V P                  P                  P                  f   V# \        V P                  P                  P                  V4       UUu/ uF  w  rVWVbK	  	  pppVP                  V P                  P                  P                  WP                  V34       V# u uppi )Nr5   rj  Ústream)r   r  Úlaunch_enter_hookrˆ  r<  r5   rj  r:   r{   r$   r0   Úlaunch_metadatarÄ   r<   rI  rõ   )rB   ÚgridrŽ  rF  Úretr5   ÚargÚarg_dicts   &&&*    r!   r�  ÚCompiledKernel.launch_metadataí  sÉ   € Ü�=‰=×*Ñ*Ò2ÙØ×ÑÔÜ˜ §	¡	¨:°t·}±}ÀhÐPVÐWÓXˆÜ˜$Ÿ(™(¤I×.Ò.°$·(±(·+±+×2MÑ2MÒ2UØˆJÜ/2°4·8±8·;±;×3HÑ3HÈ$Ô/OÔPÑ/O¡) $�D’IÑ/OˆÑPØ�‰�—‘—‘×+Ñ+¨d·M±MÀ8Ð-LÔMØˆ
ùó Qs   ÃDc                	ó@   a a€ S P                  4        R R/VV 3R llpV# )rŽ  Nc                óž  <€ V f>   \         P                  P                  4       p\         P                  P                  V4      p SP                  ! SV .VO5!  pSP
                  ! S^ ,          S^,          S^,          V SP                  SP                  V\        P                  P                  \        P                  P                  .	VO5!   R # rj   )r   rœ   r}  Úget_current_streamr�  r‹  rj  rf  r   r  r�  Úlaunch_exit_hook)rŽ  rF  rŸ   r�  r‘  rB   s   $*  €€r!   ÚrunnerÚ*CompiledKernel.__getitem__.<locals>.runnerû  s™   ø€ ØŠ~ÜŸ™×9Ñ9Ó;�ÜŸ™×9Ñ9¸&ÓA�Ø"×2Ò2°4¸ÐGÀ$ÓGˆOØ�HŠH�T˜!•W˜d 1�g t¨A¥w°¸¿¹Àt×G[ÑG[Ð]lÜ—]‘]×4Ñ4´e·m±m×6TÑ6Tð]ØW[ö]r"   )rˆ  )rB   r‘  rš  s   ff r!   Ú__getitem__ÚCompiledKernel.__getitem__ø  s*   ù€ Ø×ÑÔð	] ÷ 	]ð 	]ð ˆr"   )rk  rh  rj  rZ   ri  rõ   rö   rƒ   r„  r‚  rƒ  r5   rf  r{   N)r4   ro   rp   rq   rF   rr  rˆ  Úpropertyr‹  r�  rœ  rr   r*   r"   r!   r
  r
  —  s1   † òò<ò%vðN ñó ðò
	ör"   r
  )NNN)4Ú
__future__r   rR   rþ   Ú_C.libtritonr   r   r   Úbackends.compilerr   r   r	   r   r
   r   Úruntime.autotunerr   Úruntime.cacher   r   r   r   Úruntime.driverr   Útools.disasmr   Úpathlibr   r   rv  r¾   rÏ   rS  Úptx_prototype_patternr}   Úptx_arg_type_patternr€   r   r$   rt   Ú	lru_cacher›   r§   r»   rË   r2  r  r<  r7   rL  rW  r
  r*   r"   r!   Ú<module>rª     sî   ðÝ "Û Û ß >Ý Ý (ß 6ß !Ý .ß dÓ dÝ #Ý #Ý Û 	Û Û 	Û Û ð YÐ à	Ð ðÐ ð ,Ð à	ÐðÐ ò

÷ ñ  ÷F+ñ +ð\ ×ÒÓñOó ðOò,õ#$÷L$
ñ $
ôNI5õX÷)ñ )ô 
ˆdô 
ò÷ló lr"   