
    uki#                        d dl mZmZmZ d dlmZ d dlmZ ej                  Z
 G d de      Z ed      d        Z G d d	e      Z ed
      d        Z G d de      Z ed      d        Z G d de      Z ed      d        Z G d de      Z ed      d        Z G d de      Z ed      d        Z G d de      Z ed      d        Z G d  d!e      Z ed"      d#        Z G d$ d%e      Z ed&      d'        Z G d( d)e      Z ed*      d+        Z G d, d-e      Z ed.      d/        Z  G d0 d1e      Z! ed2      d3        Z"y4)5    )IntEnumautoIntFlag)_cext)register_attribute_builderc                   >    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZd Zy)RMWOpzDallowed 32-bit signless integer cases: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10                        	   
   c                    | t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j
                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry	| t         j                  u ry
t        d      )NandorxoraddfaddmaxminumaxuminexchzUnknown RMWOp enum entry.)r	   ANDORXORADDFADDMAXMINUMAXUMINXCHG
ValueErrorselfs    Y/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jaxlib/triton/_triton_enum_gen.py__str__zRMWOp.__str__   s    5995885995995::5995995::5::5::455    N)__name__
__module____qualname____doc__r   r    r!   r"   r#   r$   r%   r&   r'   r(   r-    r.   r,   r	   r	   
   s<    N
C	
B
C
CD
C
CDDD6r.   r	   TT_AtomicRMWAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S N    )context_ods_irIntegerAttrgetIntegerTypeget_signlessintxr8   s     r,   _tt_atomicrmwattrrB   1   9    ""7#6#6#C#CBPW#C#XZ]^_Z`aar.   c                   2    e Zd ZdZdZdZdZdZdZdZ	dZ
d	 Zy
)CacheModifierz:allowed 32-bit signless integer cases: 1, 2, 3, 4, 5, 6, 7r
   r   r   r   r   r   r   c                 "   | t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j
                  u ry| t         j                  u ry| t         j                  u ryt        d      )	Nnonecacgwbcswtcvz!Unknown CacheModifier enum entry.)	rE   NONECACGWBCSWTCVr)   r*   s    r,   r-   zCacheModifier.__str__@   s    =%%%=###=###=###=###=###=###<==r.   N)r/   r0   r1   r2   rN   rO   rP   rQ   rR   rS   rT   r-   r3   r.   r,   rE   rE   5   s-    DD	
B	
B	
B	
B	
B	
B>r.   rE   TT_CacheModifierAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_cachemodifierattrrW   S   rC   r.   c                   6    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
 Zy)DescriptorReduceKindz=allowed 32-bit signless integer cases: 1, 2, 3, 4, 5, 6, 7, 8r
   r   r   r   r   r   r   r   c                 H   | t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j
                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ryt        d	      )
Nr   r   r   incdecr   r   r   z(Unknown DescriptorReduceKind enum entry.)
rY   r"   r%   r$   INCDECr   r    r!   r)   r*   s    r,   r-   zDescriptorReduceKind.__str__c   s    '+++'+++'+++'+++'+++'+++'***'+++CDDr.   N)r/   r0   r1   r2   r"   r%   r$   r]   r^   r   r    r!   r-   r3   r.   r,   rY   rY   W   s3    G
C
C
C
C
C
C	
B
CEr.   rY   TT_DescriptorReduceKindAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_descriptorreducekindattrra   x   rC   r.   c                   "    e Zd ZdZdZdZdZd Zy)EvictionPolicy.allowed 32-bit signless integer cases: 1, 2, 3r
   r   r   c                     | t         j                  u ry| t         j                  u ry| t         j                  u ryt	        d      )Nevict_normalevict_first
evict_lastz"Unknown EvictionPolicy enum entry.)rc   NORMALEVICT_FIRST
EVICT_LASTr)   r*   s    r,   r-   zEvictionPolicy.__str__   sA    >(((!>--- >,,,=>>r.   N)r/   r0   r1   r2   ri   rj   rk   r-   r3   r.   r,   rc   rc   |   s    8FKJ?r.   rc   TT_EvictionPolicyAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_evictionpolicyattrrn      rC   r.   c                   *    e Zd ZdZdZdZdZdZdZd Z	y)	InputPrecisionz4allowed 32-bit signless integer cases: 0, 1, 2, 3, 4r   r
   r   r   r   c                     | t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j
                  u ryt        d      )Ntf32tf32x3ieeebf16x3bf16x6z"Unknown InputPrecision enum entry.)rp   TF32TF32x3IEEEBF16x3BF16x6r)   r*   s    r,   r-   zInputPrecision.__str__   sc    >&&&>(((>&&&>(((>(((=>>r.   N)
r/   r0   r1   r2   rw   rx   ry   rz   r{   r-   r3   r.   r,   rp   rp      s#    >DFDFF?r.   rp   TT_InputPrecisionAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_inputprecisionattrr~      rC   r.   c                   &    e Zd ZdZdZdZdZdZd Zy)MemSemanticz1allowed 32-bit signless integer cases: 1, 2, 3, 4r
   r   r   r   c                     | t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ryt        d      )Nrelaxedacquirereleaseacq_relzUnknown MemSemantic enum entry.)r   RELAXEDACQUIRERELEASEACQUIRE_RELEASEr)   r*   s    r,   r-   zMemSemantic.__str__   sR    ;&&&;&&&;&&&;...:;;r.   N)	r/   r0   r1   r2   r   r   r   r   r-   r3   r.   r,   r   r      s    ;GGGO	<r.   r   TT_MemSemanticAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_memsemanticattrr      rC   r.   c                   "    e Zd ZdZdZdZdZd Zy)MemSyncScoperd   r
   r   r   c                     | t         j                  u ry| t         j                  u ry| t         j                  u ryt	        d      )Ngpuctasysz Unknown MemSyncScope enum entry.)r   GPUCTASYSTEMr)   r*   s    r,   r-   zMemSyncScope.__str__   sA    <###<###<&&&;<<r.   N)r/   r0   r1   r2   r   r   r   r-   r3   r.   r,   r   r      s    8
C
CF=r.   r   TT_MemSyncScopeAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_memsyncscopeattrr      rC   r.   c                       e Zd ZdZdZdZd Zy)PaddingOptionz+allowed 32-bit signless integer cases: 1, 2r
   r   c                 d    | t         j                  u ry| t         j                  u ryt        d      )Nzeronanz!Unknown PaddingOption enum entry.)r   PAD_ZEROPAD_NANr)   r*   s    r,   r-   zPaddingOption.__str__   s0    =)))=(((<==r.   N)r/   r0   r1   r2   r   r   r-   r3   r.   r,   r   r      s    5HG>r.   r   TT_PaddingOptionAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_paddingoptionattrr      rC   r.   c                   "    e Zd ZdZdZdZdZd Zy)ProgramIDDimz.allowed 32-bit signless integer cases: 0, 1, 2r   r
   r   c                     | t         j                  u ry| t         j                  u ry| t         j                  u ryt	        d      )NrA   yzz Unknown ProgramIDDim enum entry.)r   XYZr)   r*   s    r,   r-   zProgramIDDim.__str__   s;    <>>!<>>!<>>!;<<r.   N)r/   r0   r1   r2   r   r   r   r-   r3   r.   r,   r   r      s    8	A	A	A=r.   r   TT_ProgramDimc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_programdimr     rC   r.   c                       e Zd ZdZdZdZd Zy)PropagateNanz/allowed 32-bit signless integer cases: 0, 65535r   i  c                 d    | t         j                  u ry| t         j                  u ryt        d      )NrG   allz Unknown PropagateNan enum entry.)r   rN   ALLr)   r*   s    r,   r-   zPropagateNan.__str__  s0    <$$$<###;<<r.   N)r/   r0   r1   r2   rN   r   r-   r3   r.   r,   r   r     s    9D
C=r.   r   TT_PropagateNanAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_propagatenanattrr     rC   r.   c                       e Zd ZdZdZdZd Zy)RoundingModez+allowed 32-bit signless integer cases: 0, 1r   r
   c                 d    | t         j                  u ry| t         j                  u ryt        d      )Nrtzrtnez Unknown RoundingMode enum entry.)r   RTZRTNEr)   r*   s    r,   r-   zRoundingMode.__str__  s0    <###<$$$;<<r.   N)r/   r0   r1   r2   r   r   r-   r3   r.   r,   r   r     s    5
CD=r.   r   TT_RoundingModeAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_roundingmodeattrr   (  rC   r.   c                   2    e Zd ZdZdZdZdZdZdZdZ	dZ
d	 Zy
)ScaleDotElemTypez:allowed 32-bit signless integer cases: 0, 1, 2, 3, 4, 5, 6r   r
   r   r   r   r   r   c                 "   | t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j                  u ry| t         j
                  u ry| t         j                  u ry| t         j                  u ryt        d      )	Ne4m3e5m2e2m3e3m2e2m1bf16fp16z$Unknown ScaleDotElemType enum entry.)	r   E4M3E5M2E2M3E3M2E2M1BF16FP16r)   r*   s    r,   r-   zScaleDotElemType.__str__7  s    #(((#(((#(((#(((#(((#(((#(((?@@r.   N)r/   r0   r1   r2   r   r   r   r   r   r   r   r-   r3   r.   r,   r   r   ,  s.    DDDDDDDDAr.   r   TT_ScaleDotElemTypeAttrc                     t         j                  j                  t         j                  j	                  d|      t        |             S r6   r9   r@   s     r,   _tt_scaledotelemtypeattrr   J  rC   r.   N)#enumr   r   r    jaxlib.mlir.dialects._ods_commonr   	_ods_cextjaxlib.mlir.irr   irr:   r	   rB   rE   rW   rY   ra   rc   rn   rp   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r3   r.   r,   <module>r      s  
 ( ' ? 5
,,#6G #6N ./b 0b>G >< 23b 4bE7 EB 9:b ;b?W ?$ 34b 5b?W ?0 34b 5b<' <* 01b 2b=7 =$ 12b 3b>G > 23b 4b=7 =$ O,b -b=7 = 12b 3b=7 = 12b 3bAw A< 56b 7br.   