
    Upf!                        d dl T d dl mZ d dlmZmZ d dlmZ dsdZ ed          d             Z ed	          d
             Z	 ed          d             Z
 ed          d             Z ed          d             Z ed          d             Z ed          d             Z ed          d             Z ed          d             Z ed          d             Z ed          d             Z ed          d             Z ed          d              Z ed!          d"             Z ed#          d$             Z ed%          d&             Z ed'          d(             Z ed)          d*             Z ed+          d,             Z ed-          d.             Z ed/          d0             Z ed1          d2             Z ed3          d4             Z ed5          d6             Z ed7          d8             Z  ed9          d:             Z! ed;          d<             Z" ed=          d>             Z# ed?          d@             Z$ edA          dB             Z% edC          dD             Z& edE          dF             Z' edG          dH             Z( edI          dJ             Z) edK          dL             Z* edM          dN             Z+ edO          dP             Z, edQ          dR             Z- edS          dT             Z. edU          dV             Z/ edW          dX             Z0 edY          dZ             Z1 ed[          d\             Z2 ed]          d^             Z3 ed_          d`             Z4 eda          db             Z5 edc          dd             Z6 ede          df             Z7 edg          dh             Z8	 didjl9Z: edk          dl             Z; edm          dn             Z< edo          dp             Z= edq          dr             Z>djS # e?$ r Y djS w xY w)t   )*)_GlobalDebug)register_type_casterregister_value_caster)get_dialect_registryFc                       fd}|S )Nc                 B    t                               |            | S )N)replace)AttrBuilderinsert)funckindr
   s    N/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jaxlib/mlir/ir.pydecorator_builderz5register_attribute_builder.<locals>.decorator_builder   s#    4w777     )r   r
   r   s   `` r   register_attribute_builderr      s*          r   AffineMapAttrc                 6    t                               |           S N)r   getxcontexts     r   _affineMapAttrr      s    Qr   BoolAttrc                 :    t                               | |          S Nr   )r   r   r   s     r   	_boolAttrr           <<7<+++r   DictionaryAttrc                 :    t                               | |          S r   )DictAttrr   r   s     r   	_dictAttrr%      r!   r   	IndexAttrc                 j    t                               t                              |          |           S r   )IntegerAttrr   	IndexTyper   s     r   
_indexAttrr*   #   s$    ??9===991===r   I1Attrc                 l    t                               t                              d|          |           S Nr   r   r(   r   IntegerTypeget_signlessr   s     r   _i1Attrr1   (   )    ??;33Aw3GGKKKr   I8Attrc                 l    t                               t                              d|          |           S N   r   r.   r   s     r   _i8Attrr7   -   r2   r   I16Attrc                 l    t                               t                              d|          |           S N   r   r.   r   s     r   _i16Attrr<   2   )    ??;33B3HH!LLLr   I32Attrc                 l    t                               t                              d|          |           S N    r   r.   r   s     r   _i32AttrrB   7   r=   r   I64Attrc                 l    t                               t                              d|          |           S N@   r   r.   r   s     r   _i64AttrrG   <   r=   r   SI1Attrc                 l    t                               t                              d|          |           S r-   r(   r   r/   
get_signedr   s     r   _si1AttrrL   A   )    ??;11!W1EEqIIIr   SI8Attrc                 l    t                               t                              d|          |           S r5   rJ   r   s     r   _si8AttrrP   F   rM   r   SI16Attrc                 l    t                               t                              d|          |           S r:   rJ   r   s     r   	_si16AttrrS   K   )    ??;11"g1FFJJJr   SI32Attrc                 l    t                               t                              d|          |           S r@   rJ   r   s     r   	_si32AttrrW   P   rT   r   SI64Attrc                 l    t                               t                              d|          |           S rE   rJ   r   s     r   	_si64AttrrZ   U   rT   r   UI1Attrc                 l    t                               t                              d|          |           S r-   r(   r   r/   get_unsignedr   s     r   _ui1Attrr_   Z   r2   r   UI8Attrc                 l    t                               t                              d|          |           S r5   r]   r   s     r   _ui8Attrrb   _   r2   r   UI16Attrc                 l    t                               t                              d|          |           S r:   r]   r   s     r   	_ui16Attrre   d   r=   r   UI32Attrc                 l    t                               t                              d|          |           S r@   r]   r   s     r   	_ui32Attrrh   i   r=   r   UI64Attrc                 l    t                               t                              d|          |           S rE   r]   r   s     r   	_ui64Attrrk   n   r=   r   F32Attrc                 :    t                               | |          S r   )	FloatAttrget_f32r   s     r   _f32Attrrp   s       Q000r   F64Attrc                 :    t                               | |          S r   )rn   get_f64r   s     r   _f64Attrru   x   rq   r   StrAttrc                 :    t                               | |          S r   
StringAttrr   r   s     r   _stringAttrrz   }       >>!W>---r   SymbolNameAttrc                 :    t                               | |          S r   rx   r   s     r   _symbolNameAttrr~      r{   r   SymbolRefAttrc                     t          | t                    rt                              | |          S t                              | |          S r   )
isinstancelistr   r   FlatSymbolRefAttrr   s     r   _symbolRefAttrr      sF    !T 9  G 444 $$Q$888r   r   c                 :    t                               | |          S r   )r   r   r   s     r   _flatSymbolRefAttrr            G 444r   UnitAttrc                 @    | rt                               |          S d S r   )r   r   r   s     r   	_unitAttrr      s#     ||G|,,,tr   	ArrayAttrc                 :    t                               | |          S r   r   r   r   s     r   
_arrayAttrr      s    ==G=,,,r   AffineMapArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )r   .0vr   s     r   
<listcomp>z'_affineMapArrayAttr.<locals>.<listcomp>   #    @@@.G44@@@r   r   r   s    `r   _affineMapArrayAttrr      *    ==@@@@a@@@AAAr   BoolArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )r    r   s     r   r   z"_boolArrayAttr.<locals>.<listcomp>   #    ;;;A)Aw//;;;r   r   r   s    `r   _boolArrayAttrr      *    ==;;;;;;;<<<r   DictArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )r%   r   s     r   r   z"_dictArrayAttr.<locals>.<listcomp>   r   r   r   r   s    `r   _dictArrayAttrr      r   r   FlatSymbolRefArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )r   r   s     r   r   z+_flatSymbolRefArrayAttr.<locals>.<listcomp>   s$    DDDQ,Q88DDDr   r   r   s    `r   _flatSymbolRefArrayAttrr      s*    ==DDDD!DDDEEEr   I32ArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )rB   r   s     r   r   z!_i32ArrayAttr.<locals>.<listcomp>   #    :::1(1g..:::r   r   r   s    `r   _i32ArrayAttrr      *    ==:::::::;;;r   I64ArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )rG   r   s     r   r   z!_i64ArrayAttr.<locals>.<listcomp>   r   r   r   r   s    `r   _i64ArrayAttrr      r   r   I64SmallVectorArrayAttrc                 $    t          | |          S r   r   r   s     r   _i64SmallVectorArrayAttrr      s    G,,,,r   IndexListArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   r   r   s     r   r   z'_indexListArrayAttr.<locals>.<listcomp>   s#    ???-733???r   r   r   s    `r   _indexListArrayAttrr      s*    ==????Q???@@@r   F32ArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )rp   r   s     r   r   z!_f32ArrayAttr.<locals>.<listcomp>   r   r   r   r   s    `r   _f32ArrayAttrr      r   r   F64ArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )ru   r   s     r   r   z!_f64ArrayAttr.<locals>.<listcomp>   r   r   r   r   s    `r   _f64ArrayAttrr      r   r   StrArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )rz   r   s     r   r   z!_strArrayAttr.<locals>.<listcomp>   s#    ===a+a11===r   r   r   s    `r   _strArrayAttrr      s*    ======1===>>>r   SymbolRefArrayAttrc                 P    t                               fd| D                       S )Nc                 0    g | ]}t          |          S r   )r   r   s     r   r   z'_symbolRefArrayAttr.<locals>.<listcomp>   r   r   r   r   s    `r   _symbolRefArrayAttrr      r   r   DenseF32ArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseF32ArrayAttrr      r   r   DenseF64ArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseF64ArrayAttrr      r   r   DenseI8ArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseI8ArrayAttrr      s    7333r   DenseI16ArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseI16ArrayAttrr      r   r   DenseI32ArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseI32ArrayAttrr      r   r   DenseI64ArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseI64ArrayAttrr      r   r   DenseBoolArrayAttrc                 :    t                               | |          S r   )r   r   r   s     r   _denseBoolArrayAttrr      s    !!!W!555r   TypeAttrc                 :    t                               | |          S r   r   r   r   s     r   	_typeAttrr      r!   r   TypeArrayAttrc                 <    t          fd| D                       S )Nc                 H    g | ]}t                               |           S )r   r   )r   tr   s     r   r   z"_typeArrayAttr.<locals>.<listcomp>  s)    CCCAx||Aw|77CCCr   )r   r   s    `r   _typeArrayAttrr     s(    CCCCCCCWMMMr   MemRefTypeAttrc                 "    t          | |          S r   )r   r   s     r   _memref_type_attrr   
  s    Q   r       NF64ElementsAttrc                     t                               t          j        | t          j                  t
                              |          |          S Ndtyper   typer   )DenseElementsAttrr   nparrayfloat64F64Typer   s     r   _f64ElementsAttrr     sG     $$HQbj)))W-- % 
 
 	
r   I32ElementsAttrc                     t                               t          j        | t          j                  t
                              d|          |          S )Nr   rA   r   r   )r   r   r   r   int32r/   r0   r   s     r   _i32ElementsAttrr     L     $$HQbh'''))"g)>> % 
 
 	
r   I64ElementsAttrc                     t                               t          j        | t          j                  t
                              d|          |          S )Nr   rF   r   r   )r   r   r   r   int64r/   r0   r   s     r   _i64ElementsAttrr   "  r   r   IndexElementsAttrc                     t                               t          j        | t          j                  t
                              |          |          S r   )r   r   r   r   r   r)   r   s     r   _indexElementsAttrr  *  sG     $$HQbh'''w// % 
 
 	
r   )F)@_mlir_libs._mlir.irr   _mlir_libs._mlirr   r   
_mlir_libsr   r   r   r    r%   r*   r1   r7   r<   rB   rG   rL   rP   rS   rW   rZ   r_   rb   re   rh   rk   rp   ru   rz   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   numpyr   r   r   r   r  ImportErrorr   r   r   <module>r	     s	  
 # " " " - - - - - - I I I I I I I I , , , , , ,    O,,    -,  J'', , (', ,--, , .-, K((> > )(> H%%L L &%L H%%L L &%L I&&M M '&M I&&M M '&M I&&M M '&M I&&J J '&J I&&J J '&J J''K K ('K J''K K ('K J''K K ('K I&&L L '&L I&&L L '&L J''M M ('M J''M M ('M J''M M ('M I&&1 1 '&1 I&&1 1 '&1 I&&. . '&. ,--. . .-. O,,9 9 -,9 /005 5 105 J''  (' K((- - )(- 011B B 21B O,,= = -,= O,,= = -,= 455F F 65F N++< < ,+< N++< < ,+< 566- - 76- 011A A 21A N++< < ,+< N++< < ,+< N++? ? ,+? 011B B 21B /005 5 105 /005 5 105 .//4 4 0/4 /005 5 105 /005 5 105 /005 5 105 0116 6 216 J'', , (', O,,N N -,N ,--! ! .-!$	 122
 
 32
   122
 
 32
   122
 
 32
   344
 
 54
 
 
  	 	 	DD	s   2AQ QQ