
    Upf                        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d,S )-    )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dS )RMWOpzDallowed 32-bit signless integer cases: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10                        	   
   c                 `   | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j	        u rd	S | t           j
        u rd
S 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    ^/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jaxlib/triton/_triton_enum_gen.py__str__zRMWOp.__str__   s    5955845955955:65955955:65:65:64555    N)__name__
__module____qualname____doc__r   r    r!   r"   r#   r$   r%   r&   r'   r(   r-    r.   r,   r	   r	   
   s[        NN
C	
B
C
CD
C
CDDD6 6 6 6 6r.   r	   TT_AtomicRMWAttrc                     t           j                            t           j                            d|          t          |                     S N    )context_ods_irIntegerAttrgetIntegerTypeget_signlessintxr8   s     r,   _tt_atomicrmwattrrB   1   <    ""7#6#C#CBPW#C#X#XZ]^_Z`Z`aaar.   c                   0    e Zd ZdZdZdZdZdZdZdZ	d Z
d	S )
CacheModifierz7allowed 32-bit signless integer cases: 1, 2, 3, 4, 5, 6r
   r   r   r   r   r   c                     | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS t          d          )Nnonecacgwbcswtz!Unknown CacheModifier enum entry.)rE   NONECACGWBCSWTr)   r*   s    r,   r-   zCacheModifier.__str__?   s    =%%%6=###4=###4=###4=###4=###4<===r.   N)r/   r0   r1   r2   rM   rN   rO   rP   rQ   rR   r-   r3   r.   r,   rE   rE   5   sG        AAD	
B	
B	
B	
B	
B> > > > >r.   rE   TT_CacheModifierAttrc                     t           j                            t           j                            d|          t          |                     S r6   r9   r@   s     r,   _tt_cachemodifierattrrU   P   rC   r.   c                   $    e Zd ZdZdZdZdZd ZdS )EvictionPolicy.allowed 32-bit signless integer cases: 1, 2, 3r
   r   r   c                     | t           j        u rdS | t           j        u rdS | t           j        u rdS t	          d          )Nevict_normalevict_first
evict_lastz"Unknown EvictionPolicy enum entry.)rW   NORMALEVICT_FIRST
EVICT_LASTr)   r*   s    r,   r-   zEvictionPolicy.__str__[   sJ    >(((!>>--- =>,,,<=>>>r.   N)r/   r0   r1   r2   r]   r^   r_   r-   r3   r.   r,   rW   rW   T   s8        88FKJ? ? ? ? ?r.   rW   TT_EvictionPolicyAttrc                     t           j                            t           j                            d|          t          |                     S r6   r9   r@   s     r,   _tt_evictionpolicyattrrb   f   rC   r.   c                   $    e Zd ZdZdZdZdZd ZdS )InputPrecision.allowed 32-bit signless integer cases: 0, 1, 2r   r
   r   c                     | t           j        u rdS | t           j        u rdS | t           j        u rdS t	          d          )Ntf32tf32x3ieeez"Unknown InputPrecision enum entry.)rd   TF32TF32x3IEEEr)   r*   s    r,   r-   zInputPrecision.__str__q   sJ    >&&&6>(((8>&&&6=>>>r.   N)r/   r0   r1   r2   rj   rk   rl   r-   r3   r.   r,   rd   rd   j   s8        88DFD? ? ? ? ?r.   rd   TT_InputPrecisionAttrc                     t           j                            t           j                            d|          t          |                     S r6   r9   r@   s     r,   _tt_inputprecisionattrro   |   rC   r.   c                   (    e Zd ZdZdZdZdZdZd ZdS )MemSemanticz1allowed 32-bit signless integer cases: 1, 2, 3, 4r
   r   r   r   c                     | t           j        u rdS | t           j        u rdS | t           j        u rdS | t           j        u rdS t          d          )Nrelaxedacquirereleaseacq_relzUnknown MemSemantic enum entry.)rq   RELAXEDACQUIRERELEASEACQUIRE_RELEASEr)   r*   s    r,   r-   zMemSemantic.__str__   s]    ;&&&9;&&&9;&&&9;...9:;;;r.   N)	r/   r0   r1   r2   rw   rx   ry   rz   r-   r3   r.   r,   rq   rq      s=        ;;GGGO	< 	< 	< 	< 	<r.   rq   TT_MemSemanticAttrc                     t           j                            t           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dS )MemSyncScoperX   r
   r   r   c                     | t           j        u rdS | t           j        u rdS | t           j        u rdS t	          d          )Ngpuctasysz Unknown MemSyncScope enum entry.)r   GPUCTASYSTEMr)   r*   s    r,   r-   zMemSyncScope.__str__   sJ    <###5<###5<&&&5;<<<r.   N)r/   r0   r1   r2   r   r   r   r-   r3   r.   r,   r   r      s8        88
C
CF= = = = =r.   r   TT_MemSyncScopeAttrc                     t           j                            t           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dS )PaddingOptionz+allowed 32-bit signless integer cases: 1, 2r
   r   c                 `    | t           j        u rdS | t           j        u rdS t          d          )Nzeronanz!Unknown PaddingOption enum entry.)r   PAD_ZEROPAD_NANr)   r*   s    r,   r-   zPaddingOption.__str__   s7    =)))6=(((5<===r.   N)r/   r0   r1   r2   r   r   r-   r3   r.   r,   r   r      s3        55HG> > > > >r.   r   TT_PaddingOptionAttrc                     t           j                            t           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dS )ProgramIDDimre   r   r
   r   c                     | t           j        u rdS | t           j        u rdS | t           j        u rdS t	          d          )NrA   yzz Unknown ProgramIDDim enum entry.)r   XYZr)   r*   s    r,   r-   zProgramIDDim.__str__   sG    <>!!3<>!!3<>!!3;<<<r.   N)r/   r0   r1   r2   r   r   r   r-   r3   r.   r,   r   r      s8        88	A	A	A= = = = =r.   r   TT_ProgramDimc                     t           j                            t           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dS )PropagateNanz/allowed 32-bit signless integer cases: 0, 65535r   i  c                 `    | t           j        u rdS | t           j        u rdS t          d          )NrG   allz Unknown PropagateNan enum entry.)r   rM   ALLr)   r*   s    r,   r-   zPropagateNan.__str__   s7    <$$$6<###5;<<<r.   N)r/   r0   r1   r2   rM   r   r-   r3   r.   r,   r   r      s3        99D
C= = = = =r.   r   TT_PropagateNanAttrc                     t           j                            t           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dS )RoundingModez+allowed 32-bit signless integer cases: 0, 1r   r
   c                 `    | t           j        u rdS | t           j        u rdS t          d          )Nrtzrtnez Unknown RoundingMode enum entry.)r   RTZRTNEr)   r*   s    r,   r-   zRoundingMode.__str__   s7    <###5<$$$6;<<<r.   N)r/   r0   r1   r2   r   r   r-   r3   r.   r,   r   r      s3        55
CD= = = = =r.   r   TT_RoundingModeAttrc                     t           j                            t           j                            d|          t          |                     S r6   r9   r@   s     r,   _tt_roundingmodeattrr      rC   r.   N)enumr   r   r    jaxlib.mlir.dialects._ods_commonr   	_ods_cextjaxlib.mlir.irr   irr:   r	   rB   rE   rU   rW   rb   rd   ro   rq   r}   r   r   r   r   r   r   r   r   r   r   r3   r.   r,   <module>r      s  
 ( ' ' ' ' ' ' ' ' ' ? ? ? ? ? ? 5 5 5 5 5 5
,#6 #6 #6 #6 #6G #6 #6 #6N .//b b 0/b> > > > >G > > >6 233b b 43b? ? ? ? ?W ? ? ?$ 344b b 54b? ? ? ? ?W ? ? ?$ 344b b 54b< < < < <' < < <* 011b b 21b= = = = =7 = = =$ 122b b 32b> > > > >G > > > 233b b 43b= = = = =7 = = =$ O,,b b -,b= = = = =7 = = = 122b b 32b= = = = =7 = = = 122b b 32b b br.   