
    Wh3"                        d dl T d dl mZ d dlmZmZ d dlmZ dud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 edi          dj             Z9	 dkdll:Z; edm          dn             Z< edo          dp             Z= edq          dr             Z> eds          dt             Z?dlS # e@$ r Y dlS w xY w)v   )*)_GlobalDebug)register_type_casterregister_value_caster)get_dialect_registryFc                       fd}|S )Nc                 B    t                               |            | S )N)replace)AttrBuilderinsert)funckindr
   s    T/var/www/html/movieo_spanner_bot/venv/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   IntegerSetAttrc                 6    t                               |           S r   )r   r   r   s     r   _integerSetAttrr      s    a   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   _i1Attrr4   -   )    ??;33Aw3GGKKKr   I8Attrc                 l    t                               t                              d|          |           S N   r"   r1   r   s     r   _i8Attrr:   2   r5   r   I16Attrc                 l    t                               t                              d|          |           S N   r"   r1   r   s     r   _i16Attrr?   7   )    ??;33B3HH!LLLr   I32Attrc                 l    t                               t                              d|          |           S N    r"   r1   r   s     r   _i32AttrrE   <   r@   r   I64Attrc                 l    t                               t                              d|          |           S N@   r"   r1   r   s     r   _i64AttrrJ   A   r@   r   SI1Attrc                 l    t                               t                              d|          |           S r0   r+   r   r2   
get_signedr   s     r   _si1AttrrO   F   )    ??;11!W1EEqIIIr   SI8Attrc                 l    t                               t                              d|          |           S r8   rM   r   s     r   _si8AttrrS   K   rP   r   SI16Attrc                 l    t                               t                              d|          |           S r=   rM   r   s     r   	_si16AttrrV   P   )    ??;11"g1FFJJJr   SI32Attrc                 l    t                               t                              d|          |           S rC   rM   r   s     r   	_si32AttrrZ   U   rW   r   SI64Attrc                 l    t                               t                              d|          |           S rH   rM   r   s     r   	_si64Attrr]   Z   rW   r   UI1Attrc                 l    t                               t                              d|          |           S r0   r+   r   r2   get_unsignedr   s     r   _ui1Attrrb   _   r5   r   UI8Attrc                 l    t                               t                              d|          |           S r8   r`   r   s     r   _ui8Attrre   d   r5   r   UI16Attrc                 l    t                               t                              d|          |           S r=   r`   r   s     r   	_ui16Attrrh   i   r@   r   UI32Attrc                 l    t                               t                              d|          |           S rC   r`   r   s     r   	_ui32Attrrk   n   r@   r   UI64Attrc                 l    t                               t                              d|          |           S rH   r`   r   s     r   	_ui64Attrrn   s   r@   r   F32Attrc                 :    t                               | |          S r!   )	FloatAttrget_f32r   s     r   _f32Attrrs   x       Q000r   F64Attrc                 :    t                               | |          S r!   )rq   get_f64r   s     r   _f64Attrrx   }   rt   r   StrAttrc                 :    t                               | |          S r!   
StringAttrr   r   s     r   _stringAttrr}          >>!W>---r   SymbolNameAttrc                 :    t                               | |          S r!   r{   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   )rE   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   )rJ   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   )rs   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   )rx   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   )r}   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   rD   r"   r   )r   r   r   r   int32r2   r3   r   s     r   _i32ElementsAttrr     L     $$HQbh'''))"g)>> % 
 
 	
r   I64ElementsAttrc                     t                               t          j        | t          j                  t
                              d|          |          S )Nr   rI   r"   r   )r   r   r   r   int64r2   r3   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)A_mlir_libs._mlir.irr   _mlir_libs._mlirr   r   
_mlir_libsr   r   r   r   r#   r(   r-   r4   r:   r?   rE   rJ   rO   rS   rV   rZ   r]   rb   re   rh   rk   rn   rs   rx   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   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   AQ Q%$Q%