Ë
    Ž”vh	  ã                   ó”   — 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 d dlmZ d dlmZmZmZ d dlmZmZmZ d dlmZmZmZmZmZmZmZ y)é    )ÚIndexÚ	MapTracerÚMeshAxisNameÚMeshComputationÚMeshExecutableÚPmapExecutableÚglobal_aval_to_result_handlerÚglobal_avals_to_results_handlerÚglobal_result_handlersÚparallel_callableÚ
shard_argsÚ
xla_pmap_p)Úthread_resources)Úare_op_shardings_equalÚis_op_sharding_replicatedÚop_sharding_to_indices)ÚArrayMappingÚUNSPECIFIEDÚarray_mapping_to_axis_resources)ÚChunkedÚ
NoShardingÚ
ReplicatedÚShardedAxisÚShardingSpecÚ	UnstackedÚspec_to_indicesN) Újax._src.interpreters.pxlar   r   r   r   r   r   r	   r
   r   r   r   r   Újax._src.meshr   Újax._src.op_shardingsr   r   r   Újax._src.sharding_implsr   r   Ú_UNSPECIFIEDr   Újax._src.sharding_specsr   r   r   r   r   r   r   © ó    úP/opt/face_recognition/venv/lib/python3.12/site-packages/jax/interpreters/pxla.pyú<module>r&      sA   ð÷÷ ÷ ó õ÷ñ ÷ñ ÷÷ ò r$   