
    uki                          d dl mZmZ ddl ddlmZ ddlmZ 	 ddl ddlm	Z
  e
j                  ed
       G d de             Zed	d	d	d	d	ddeeeef      fd       Zy	# e$ rZ ed      ed	Z[ww xY w)    )DictOptional   )*)_Dialect   )	region_op)_cextz+Error loading imports from extension moduleNT)replacec                   :     e Zd ZdZddd fd
Zed        Z xZS )ModuleOpz'Specialization for the module op class.Nlocipc                v    t         |   ||       | j                  d   j                  j	                         }y )Nr   r   )super__init__regionsblocksappend)selfr   r   body	__class__s       W/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jaxlib/mlir/dialects/builtin.pyr   zModuleOp.__init__   s1    SR(||A%%,,.    c                 :    | j                   d   j                  d   S )Nr   )r   r   )r   s    r   r   zModuleOp.body   s    ||A%%a((r   )__name__
__module____qualname____doc__r   propertyr   __classcell__)r   s   @r   r   r      s$    1"t / ) )r   r   )sym_namesym_visibilityattrsr   r   r%   c                     t         j                  | |||      }|i }|j                         D ]  \  }}||j                  j                  |<     |S )N)r#   r$   r   r   )r   __base__items	operation
attributes)r#   r$   r%   r   r   mod	attr_nameattrs           r   moduler.      sb     

.cb  C } ;;= 3	4.2  +3 Jr   )typingr   r   _builtin_ops_genr   extras.metar	   ir_ods_commonr
   	_ods_cextImportErroreRuntimeErrorregister_operationr   str	Attributer.    r   r   <module>r<      s   
 "  & #M/
 h5	)x 	) 6	)  ,0 Di()	 %  M
D
E1LMs   A# #A6(	A11A6