
    h8              	         % S SK Jr  S SKrS SKrS SKrS SKrS SKrS SKJr  S SK	J
r
  S SKJrJrJrJrJrJrJrJrJrJrJrJr  \(       a  SSKJrJr  SSKJrJr  SS	KJrJ r J!r!   " S
 S5      r"\"" 5       r#Sr$S\%S'   SOS jr&SPS jr'SQS jr(\" S5      r)\" S5      r* " S S\\)\*4   5      r+ " S S\+\,\,4   5      r- " S S\+\.\.4   5      r/ " S S\+\0\04   5      r1 " S S\\*\)4   \+\\*   \\)   4   5      r2\" S5      r3 " S S \\3   \2\\3   \\3   4   5      r4\ " S! S"5      5       r5 " S# S$\+\,\54   5      r6 " S% S&\2\,\,4   \-5      r7 " S' S(\2\.\.4   \/5      r8\" SS)9 " S* S+5      5       r9 " S, S-\5      r:\" S.S/S09r; " S1 S/5      r< " S2 S3\5      r= " S4 S5\<5      r> " S6 S7\<5      r?S8\%S9'    " S: S8\<5      r@ " S; S<\<5      rA " S= S>\<5      rB " S? S@\5      rC\D\\,\04      rE " SA SB\5      rF " SC SD\5      rG " SE SF\<5      rH " SG SH\<5      rI " SI SJ\<5      rJ " SK SL\<5      rK " SM SN\<5      rL\>" 5       rM\?" 5       rN\@" 5       rO\A" 5       rP\B" 5       rQ\H" 5       rR\I" 5       rS\J" 5       rT\K" 5       rU\L" 5       rVg)R    )annotationsN)	dataclass)contextmanager)castAnyCallable	GeneratorGenericOptionalProtocolTypeTypeVar	TypedDictTYPE_CHECKINGUnion   )CacheManagerRemoteCacheBackend)JitFunctionInfoKernelParam)	ASTSourceLazyDictIRSourcec                      \ rS rSrSrg)Env    N__name__
__module____qualname____firstlineno____static_attributes__r       F/var/www/fran/franai/venv/lib/python3.13/site-packages/triton/knobs.pyr   r          r$   r   Tboolpropagate_envc                X    [         R                  " U 5      nUb  UR                  5       $ U$ N)osgetenvstrip)keyress     r%   r,   r,      s$    
))C.C/399;2s2r$   c                    [         (       d  g Ub  U[        R                  U '   g U [        R                  ;   a  [        R                  U 	 g g r*   )r(   r+   environ)r.   values     r%   setenvr3   !   s9    =

3	

	JJsO 
r$   c                    U c  g[        U 5      nU[        L a  U (       a  S4$ S4$ U[        L a  U 4$ U[        L a  [        U 5      4$ g )Nr*   10)typer'   strint)valts     r%   toenvr<   +   sQ    
{S	ADy&&&&CxwCxC|r$   SetTypeGetTypec                  x    \ rS rSrSS jrSS jrSS jr\SS j5       rSS jr	SS jr
SS jrSS	 jrSS
 jrSrg)env_baseB   c                X   ^ Xl         [        T5      (       a  TU l        g U4S jU l        g )Nc                    > T $ r*   r   )defaults   r%   <lambda>#env_base.__init__.<locals>.<lambda>F   s    X_r$   )r.   callablerD   )selfr.   rD   s     `r%   __init__env_base.__init__D   s     9A'9J9JgP_r$   c                    X l         g r*   )name)rH   objclassrL   s      r%   __set_name__env_base.__set_name__H   s    	r$   c                    Uc  [        S[        U 5       S35      eU R                  UR                  ;   a(  U R	                  UR                  U R                     5      $ U R                  5       $ )NzCannot access z on non-instance)AttributeErrorr7   rL   __dict__	transformget)rH   objrM   s      r%   __get__env_base.__get__K   sX    ; >$t*=M!NOO99$>>#,,tyy"9::88:r$   c                ,    [        U R                  5      $ r*   )r,   r.   rH   s    r%   env_valenv_base.env_valT   s    dhhr$   c                    U R                   nU R                  Uc  U R                  5       5      $ U R                  U5      5      $ r*   )rZ   rS   rD   from_env)rH   envs     r%   rT   env_base.getX   s5    ll~~dllnTTsASTTr$   c                   [        U[        5      (       a'  UR                  R                  U R                  S 5        g X!R                  U R                  '   [        U5      =n(       a  [        U R                  US   5        g g )Nr   )
isinstancer   rR   poprL   r<   r3   r.   )rH   rU   r2   rZ   s       r%   __set__env_base.__set__\   s]    eS!!LLTYY-&+LL#,&w&txx, 'r$   c                P    UR                   R                  U R                  S 5        g r*   )rR   rb   rL   )rH   rU   s     r%   
__delete__env_base.__delete__d   s    D)r$   c                "    [        [        U5      $ r*   )r   r>   rH   r:   s     r%   rS   env_base.transformg   s     GS!!r$   c                    [        5       er*   )NotImplementedErrorri   s     r%   r]   env_base.from_envl   s    !##r$   )rD   r.   rL   N)r.   r8   rD   z%Union[SetType, Callable[[], SetType]]returnNone)rM   zType[object]rL   r8   rn   ro   )rU   zOptional[object]rM   zOptional[Type[object]]rn   r>   )rn   z
str | None)rn   r>   )rU   objectr2   zUnion[SetType, Env]rn   ro   )rU   rp   rn   ro   )r:   r=   rn   r>   )r:   r8   rn   r=   )r   r    r!   r"   rI   rN   rV   propertyrZ   rT   rc   rf   rS   r]   r#   r   r$   r%   r@   r@   B   s@    `    U-*"
$r$   r@   c                      \ rS rSrSS jrSrg)env_strp   c                    U$ r*   r   ri   s     r%   r]   env_str.from_envr       
r$   r   Nr:   r8   rn   r8   )r   r    r!   r"   r]   r#   r   r$   r%   rs   rs   p   s    r$   rs   c                  :   ^  \ rS rSrSSU 4S jjjrSS jrSrU =r$ )env_boolv   c                $   > [         TU ]  X5        g r*   superrI   rH   r.   rD   	__class__s      r%   rI   env_bool.__init__x       &r$   c                (    UR                  5       S;   $ )N)r5   trueyesony)lowerri   s     r%   r]   env_bool.from_env{   s    yy{===r$   r   )F)r.   r8   rD   zUnion[bool, Callable[[], bool]]rn   ro   )r:   r8   rn   r'   r   r    r!   r"   rI   r]   r#   __classcell__r   s   @r%   rz   rz   v   s    ' '> >r$   rz   c                  :   ^  \ rS rSrSSU 4S jjjrSS jrSrU =r$ )env_int   c                $   > [         TU ]  X5        g r*   r}   r   s      r%   rI   env_int.__init__   r   r$   c                x     [        U5      $ ! [         a"  n[        SU R                   SU S35      UeS nAff = f)NzUnable to use =z: expected int)r9   
ValueErrorRuntimeErrorr.   )rH   r:   excs      r%   r]   env_int.from_env   sD    	Ys8O 	Yz3%~NOUXX	Ys   
 
949r   )r   )r.   r8   rD   zUnion[int, Callable[[], int]]rn   ro   )r:   r8   rn   r9   r   r   s   @r%   r   r      s    ' 'Y Yr$   r   c                  ,   ^  \ rS rSrSU 4S jjrSrU =r$ )env_opt_base   c                &   > [         TU ]  US 5        g r*   r}   )rH   r.   r   s     r%   rI   env_opt_base.__init__   s    d#r$   r   )r.   r8   rn   ro   )r   r    r!   r"   rI   r#   r   r   s   @r%   r   r      s    $ $r$   r   	ClassTypec                  6   ^  \ rS rSrSU 4S jjrSS jrSrU =r$ )	env_class   c                0   > [         TU ]  U5        X l        g r*   )r~   rI   r7   )rH   r.   r7   r   s      r%   rI   env_class.__init__   s    	r$   c           	       ^  UR                  SS5      n[        U5      S:w  a  [        ST R                   SU S35      e[	        [
        R                  " US   5      US   5      n[        U 4S jUR                  5        5       5      (       d)  [        S	U S
T R                   ST R                   S35      e[        [        [           U5      $ )N:r      zUnable to read z: 'z ' isn't of the form MODULE:CLASSr   c              3  T   >#    U  H  oR                   TR                  :H  v   M     g 7fr*   )r   r7   ).0crH   s     r%   	<genexpr>%env_class.from_env.<locals>.<genexpr>   s     ?YJJ$))+Ys   %(zUnable to use 'z' from z: not of type '')splitlenr   r.   getattr	importlibimport_moduleanymror7   r   r   r   )rH   r:   compsclss   `   r%   r]   env_class.from_env   s    		#q!u:?
#cUBbcddi--eAh7qB?SWWY?@@WTXXJoVZV_V_U``abccDOS))r$   )r7   )r.   r8   r7   r8   rn   ro   )r:   r8   rn   zType[ClassType]r   r   s   @r%   r   r      s    
	* 	*r$   r   c                  >    \ rS rSr% S\S'   S\S'   \SS j5       rSrg)	
NvidiaTool   r8   pathversionc                ,    [         R                  " U S/[         R                  S9nUc  g [        R                  " SUR                  S5      [        R                  S9nUc  g [        XR                  S5      5      $ ! [         R                   a     g f = f)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_pathNvidiaTool.from_path   s    		,,dK-@IZIZ[F~ii 96==;QY[YeYefGdMM!$455,, 		s   (A< 6A< "A< <BBr   N)r   r8   rn   zOptional[NvidiaTool])r   r    r!   r"   __annotations__staticmethodr   r#   r   r$   r%   r   r      s    
IL
 
r$   r   c                  @   ^  \ rS rSrSU 4S jjrSS jrSS jrSrU =r$ )	env_nvidia_tool   c                   >^  U[         R                  " S5      -  nUT l        [        TT ]  SUR                  5        S3U 4S j5        g )NEXETRITON__PATHc                    > [         R                  R                  [         R                  R                  [        5      SSST R
                  5      $ )Nbackendsnvidiabin)r+   r   joindirname__file__binaryrY   s   r%   rE   *env_nvidia_tool.__init__.<locals>.<lambda>   s3    "'',,GGOOH%KKC
r$   )	sysconfigget_config_varr   r~   rI   upper)rH   r   r   s   ` r%   rI   env_nvidia_tool.__init__   sC    )**51176<<>"2%8 ;
 	r$   c                   UU R                  5       /nU HV  nU(       a*  [        R                  " U[        R                  5      (       d  M6  [        R                  U5      =n(       d  MT  Us  $    [        SU R                   35      e)NzCannot find )rD   r+   accessX_OKr   r   r   r   )rH   r   pathstools       r%   rS   env_nvidia_tool.transform   sp     LLN	
 Dryyrww77!++D11t1	  \$++788r$   c                    U$ r*   r   ri   s     r%   r]   env_nvidia_tool.from_env   rw   r$   )r   )r   r8   rn   ro   )r   r8   rn   r   rx   )	r   r    r!   r"   rI   rS   r]   r#   r   r   s   @r%   r   r      s    	9 r$   r   c                      \ rS rSrSrg)env_opt_str   r   Nr   r   r$   r%   r   r      r&   r$   r   c                      \ rS rSrSrg)env_opt_bool   r   Nr   r   r$   r%   r   r      r&   r$   r   )frozenc                  `    \ rS rSr% SrS\S'   S\S'   S\S'   \SS j5       r\SS	 j5       rS
r	g)CompileTimes   za
Model holding timing information for an invocation of the compiler.

All times in microseconds.
r9   ir_initializationzlist[tuple[str, int]]lowering_stagesstore_resultsc                :    [        S U R                   5       5      $ )Nc              3  *   #    U  H	  oS    v   M     g7f)r   Nr   )r   stages     r%   r   .CompileTimes.total_lowering.<locals>.<genexpr>   s     ?*>!H*>s   )sumr   rY   s    r%   total_loweringCompileTimes.total_lowering   s    ?$*>*>?@@r$   c                N    U R                   U R                  -   U R                  -   $ r*   )r   r   r   rY   s    r%   totalCompileTimes.total   s$    %%(;(;;d>P>PPPr$   r   N)rn   r9   )
r   r    r!   r"   __doc__r   rq   r   r   r#   r   r$   r%   r   r      sI      +* A A Q Qr$   r   c                  *    \ rS rSr      SS jrSrg)CompilationListener   c                   g r*   r   )rH   srcmetadatametadata_grouptimes	cache_hits         r%   __call__CompilationListener.__call__       r$   r   N)r  zUnion[ASTSource, IRSource]r  dict[str, Any]r  zdict[str, str]r  r   r  r'   rn   ro   r   r    r!   r"   r	  r#   r   r$   r%   r  r     s    $15:>r$   r  
knobs_type
base_knobs)boundc                  d    \ rS rSr\S	S j5       r\S
S j5       rSS jrSS jr\	SS j5       r
Srg)r  i  c                    [        U 5      R                  R                  5        VVs0 s H  u  p[        U[        5      (       d  M  X_M      snn$ s  snnf r*   )r7   rR   itemsra   r@   )rH   kvs      r%   knob_descriptorsbase_knobs.knob_descriptors
  sN    
 T
++113
 4!X& AD3
 	
 
s   AAc                t    U R                   R                  5        Vs0 s H  o[        X5      _M     sn$ s  snf r*   )r  keysr   )rH   r  s     r%   knobsbase_knobs.knobs  s3    -1-B-B-G-G-IJ-I74##-IJJJs   5c                p    [        U 5      " 5       nUR                  R                  U R                  5        U$ r*   )r7   rR   update)rH   r/   s     r%   copybase_knobs.copy  s(    4jlDMM*
r$   c                ^    U R                   R                  5        H  n[        X5        M     U $ r*   )r  r  delattr)rH   knobs     r%   resetbase_knobs.reset  s(    ))..0DD 1r$   c              #    #     U R                   R                  5        Vs0 s H  oR                  UR                  _M     nn[	        U R
                  5      nS v   U R
                  R                  5         U R
                  R                  U5        UR                  5        HD  u  pEUb  U[        R                  U'   M  U[        R                  ;   d  M3  [        R                  U	 MF     g s  snf ! U R
                  R                  5         U R
                  R                  W5        WR                  5        HD  u  pEUb  U[        R                  U'   M  U[        R                  ;   d  M3  [        R                  U	 MF     f = f7fr*   )r  valuesr.   rZ   dictrR   clearr  r  r+   r1   )rH   r"  initial_envorigr  r  s         r%   scopebase_knobs.scope!  s    	&>B>S>S>Z>Z>\]>\d88T\\1>\K]&DMM!MM  &#))+=$%BJJqM"**_

1	 , ^ MM!MM  &#))+=$%BJJqM"**_

1	 ,s;   F C.  C) C. A4F F )C. .A5E='E==F r   N)rn   zdict[str, env_base])rn   r  )rH   r  rn   r  )rn   zGenerator[None, None, None])r   r    r!   r"   rq   r  r  r  r#  r   r+  r#   r   r$   r%   r  r    sH    
 
 K K

 & &r$   c                  &    \ rS rSr    SS jrSrg)	BuildImpli2  c                   g r*   r   )rH   rL   r  srcdirlibrary_dirsinclude_dirs	librariess          r%   r	  BuildImpl.__call__4  r  r$   r   N)rL   r8   r  r8   r0  r8   r1  	list[str]r2  r5  r3  r5  rn   r8   r  r   r$   r%   r.  r.  2  s    %-0r$   r.  c                      \ rS rSr% Sr\" S5      rS\S'   \" S5      rS\S'   \" S5      r	S\S	'   S
r
S\S'   \SS j5       rSrg
)build_knobsi9  z<Configuration controlling how the native compiler is invokedCCr   ccTRITON_CUDACRT_PATHcudacrt_pathTRITON_CUDART_PATHcudart_pathNzOptional[BuildImpl]implc                f    U R                   U R                  4 Vs1 s H	  oc  M  UiM     sn$ s  snf r*   )r;  r=  )rH   r   s     r%   backend_dirsbuild_knobs.backend_dirsB  s-    "&"3"3T5E5E!F[!F!F[[[s   ..r   )rn   zset[str])r   r    r!   r"   r   r   r9  r   r;  r=  r>  rq   r@  r#   r   r$   r%   r7  r7  9  sS    F!$'B' +,A BL+B*+?@K@ $D
$\ \r$   r7  c                  j    \ rS rSr% \" SS5      rS\S'   \" SS5      rS\S'   \" S	S
5      r	S\S'   Sr
g)redis_knobsiG  TRITON_REDIS_KEY_FORMATztriton:{key}:{filename}rs   
key_formatTRITON_REDIS_HOST	localhosthostTRITON_REDIS_PORTi  r   portr   N)r   r    r!   r"   rs   rE  r   rH  r   rJ  r#   r   r$   r%   rC  rC  G  s9    !";=VWJW/=D'=/6D'6r$   rC  cache_knobscachec                      \ rS rSr% \" SS 5      rS\S'   \" SS 5      rS\S'   \" S	S
 5      rS\S'   \" SS 5      r	S\S'   \
" SS5      rS\S'   \
" SS5      rS\S'   SS jrSrg)rK  iP  TRITON_HOMEc                 @    [         R                  R                  S5      $ )Nz~/)r+   r   
expanduserr   r$   r%   rE   cache_knobs.<lambda>Q  s    rww7I7I$7Or$   rs   home_dirTRITON_DUMP_DIRc                 ,    [         R                  S5      $ )NdumprL  get_triton_dirr   r$   r%   rE   rQ  S  s    5;O;OPV;Wr$   dump_dirTRITON_OVERRIDE_DIRc                 ,    [         R                  S5      $ )NoverriderV  r   r$   r%   rE   rQ  T  s    5CWCWXbCcr$   override_dirTRITON_CACHE_DIRc                 ,    [         R                  S5      $ )NrL  rV  r   r$   r%   rE   rQ  U  s    u7K7KG7Tr$   dirTRITON_CACHE_MANAGERr   zenv_class[CacheManager]manager_classTRITON_REMOTE_CACHE_BACKENDr   zenv_class[RemoteCacheBackend]remote_manager_classc                X    [         R                  R                  U R                  SU5      $ )Nz.triton)r+   r   r   rR  )rH   r   s     r%   rW  cache_knobs.get_triton_dirZ  s    ww||DMM9g>>r$   r   N)r   r8   rn   r8   )r   r    r!   r"   rs   rR  r   rX  r\  r_  r   ra  rc  rW  r#   r   r$   r%   rK  rK  P  sz    /OPHgP 13WXHgX#$9;cdL'd-/TUCU-67M~-^M*^:CDacw:x7x?r$   c                     \ rS rSr% \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S	5      r	S\S
'   \
" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   SrS\S'   Srg)compilation_knobsi^  TRITON_KERNEL_OVERRIDErz   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]listenerr   )r   r    r!   r"   rz   r[  r   rj  rl  rn  r   rp  rr  rt  rv  rx  ry  r#   r   r$   r%   rg  rg  ^  s    !":;Hh; !56GX6"*+E"FxF'(?@NH@),7J7$%9:K:"*+E"FxF$,-I$JJ,45Y,ZZ15H.5r$   rg  c                  J    \ rS rSr% \" S5      rS\S'   \" S5      rS\S'   Srg)	autotuning_knobsil  TRITON_CACHE_AUTOTUNINGrz   rL  TRITON_PRINT_AUTOTUNINGprintr   N)	r   r    r!   r"   rz   rL  r   r~  r#   r   r$   r%   r{  r{  l  s"    89E8989E89r$   r{  c                      \ rS rSrSS jrSrg)
LaunchHookiq  c                    g r*   r   )rH   r  s     r%   r	  LaunchHook.__call__s  s    r$   r   N)r  r   rn   ro   r  r   r$   r%   r  r  q  s    r$   r  c                      \ rS rSr% S\S'   S\S'   S\S'   S\S	'   S\S
'   S\S'   S\S'   S\S'   S\S'   S\S'   S\S'   S\S'   S\S'   Srg)JITHookCompileInfoi|  r8   r.   zdict[KernelParam, str]	signaturer9   devicero   	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_warmupr   N)r   r    r!   r"   r   r#   r   r$   r%   r  r  |  sI    	H%%KONMO!!,,::Or$   r  c                  &    \ rS rSr    SS jrSrg)JITHooki  c                   g r*   r   )rH   r.   reprfncompileis_manual_warmupalready_compileds          r%   r	  JITHook.__call__  r  r$   r   N)r.   r8   r  r8   r  r   r  r  r  r'   r  r'   rn   zOptional[bool]r  r   r$   r%   r  r    s    #',:r$   r  c                      \ rS rSr% \" S5      rS\S'   \" S5      rS\S'   \" S5      r	S\S	'   S
r
S\S'   S
rS\S'   S
rS\S'   S
rS\S'   Srg
)runtime_knobsi  TRITON_INTERPRETrz   	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_hookr   )r   r    r!   r"   rz   r  r   r  r   r  r  r  r  r  r#   r   r$   r%   r  r    sd    "#56Ix6~.E8.!,-C!DM;D.2+2-1*1 )-N%, 04,3r$   r  c                  L    \ rS rSr% \" S5      rS\S'   \" SS5      rS\S'   S	r	g
)language_knobsi  TRITON_F32_DEFAULTr   fp32_defaultTRITON_DEFAULT_FP_FUSIONTrz   default_fp_fusionr   N)
r   r    r!   r"   r   r  r   rz   r  r#   r   r$   r%   r  r    s%     +,@ AL+A"*+Et"LxLr$   r  c                      \ rS rSr% \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \	" S5      r
S\S'   \	" S	5      rS\S
'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   Srg)nvidia_knobsi  	cuobjdumpr   nvdisasmptxasNVPTX_ENABLE_DUMPrz   
dump_nvptxDISABLE_PTXAS_OPTdisable_ptxas_optTRITON_MOCK_PTX_VERSIONr   mock_ptx_versionTRITON_LIBDEVICE_PATHlibdevice_pathTRITON_LIBCUDA_PATHlibcuda_pathr   N)r   r    r!   r"   r   r  r   r  r  rz   r  r  r   r  r  r  r#   r   r$   r%   r  r    s    !0!=I= /
 ;Ho;,W5E?5#$78J8"*+>"?x?$/0I$JkJ"-.E"FNKF +,A BL+Br$   r  c                     \ rS rSr% \" S5      rS\S'   \" S5      rS\S'   \" S5      r	S\S	'   \" S
5      r
S\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   \" S5      rS\S'   Srg)	amd_knobsi  AMDGCN_USE_BUFFER_OPSrz   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_fopsr   N)r   r    r!   r"   rz   r  r   r  r   r  r  r   r  r  r   r  r  r  r  r#   r   r$   r%   r  r    s    '(?@NH@$%9:K:*+?@K@'(=>Hk> (44S'TT,89],^\^&'CDOWD%&ABNGB'(CDNHD&./M&N8Nr$   r  c                  0    \ rS rSr% \" S5      rS\S'   Srg)proton_knobsi  TRITON_CUPTI_LIB_PATHr   	cupti_dirr   N)r   r    r!   r"   r   r  r   r#   r   r$   r%   r  r    s    ()@AI{Ar$   r  )r.   r8   rn   Optional[str])r.   r8   r2   r  rn   ro   )r:   r   rn   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   r^   r(   r   r,   r3   r<   r=   r>   r@   r8   rs   r'   rz   r9   r   r   r   r   r   r   r   r   r   r  r  r  r.  r7  rC  rK  rg  r{  r  list
KernelAttrr  r  r  r  r  r  r  buildredisrL  compilation
autotuningruntimelanguager   amdprotonr   r$   r%   <module>r     s   "  	 	   ! % ~ ~ ~ ~?9@@	 	 
et 3
& )

)
+$ww'( +$\hsCx  >xd
# >	YhsCx  	Y$77G+,hx7H(SZJ[7[.\ $ K 	*	"Li$y/1Q$R *&   $hsJ/ B	,sCx(' 		<d
+X 	 $Q Q Q4(  \6
'& '&T \* \7* 7  ?* ?6
 6:z :
  %S/"
  h 4J 4MZ M

C: 
CO
 O B: B 	!

/	k	r$   