Ë
    ñukiî  ã                   óL  — d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. d dl/m0Z0 d dl1m2Z2m3Z3m4Z4m5Z5 d dl6m7Z7 d dl8m9Z9 ddiZ:d dl;Z<e<jz                  r	 [<yd d	l>m?Z@  e@eAe:«      ZB[@[<y)
é    )-ÚAxisContextÚConstantHandlerÚDEVICE_TO_DEVICE_TYPEÚLoweringParametersÚLoweringResultÚLoweringRuleÚLoweringRuleContextÚModuleContextÚRECV_FROM_HOST_TYPEÚSEND_TO_HOST_TYPEÚShapePolyLoweringStateÚTokenÚTokenSetÚValueÚcall_loweringÚ
_loweringsÚ_platform_specific_loweringsÚaval_to_ir_typeÚaval_to_ir_typesÚcore_call_loweringÚdense_bool_elementsÚdense_int_arrayÚdense_int_elementsÚdtype_to_ir_typeÚflatten_ir_typesÚflatten_ir_valuesÚunflatten_ir_values_like_typesÚi32_attrÚi64_attrÚirÚir_attributeÚir_constantÚir_type_handlersÚjaxpr_subcompÚ	lower_funÚlower_jaxpr_to_funÚlower_jaxpr_to_moduleÚmake_ir_contextÚmerge_mlir_modulesÚmodule_to_bytecodeÚmodule_to_stringÚregister_constant_handlerÚregister_loweringÚshape_tensorÚ
token_type)ÚMesh)ÚMeshAxisNameÚReplicaAxisContextÚSPMDAxisContextÚShardingContext)Úlowerable_effects)Úemit_python_callbackÚcustom_call)zUmlir.custom_call was removed in JAX v0.8.0; use the APIs provided by jax.ffi instead.NN)Údeprecation_getattr)CÚjax._src.interpreters.mlirr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   Ú_call_loweringr   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.   r/   Újax._src.meshr0   Újax._src.sharding_implsr1   r2   r3   r4   Újax._src.effectsr5   Újax._src.callbackr6   Ú_deprecationsÚtypingÚ_typingÚTYPE_CHECKINGÚjax._src.deprecationsr8   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úP/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/interpreters/mlir.pyú<module>rJ      s›   ð÷.÷ .÷ .÷ .÷ .÷ .÷ .÷ .÷ .÷ .÷ .õ .õ` '÷ó õ Dõ
ð ð ð	€ó à
×ÒØñ õ	 Pá$ X¨}Ó=€+ØÙrH   