Ë
    ñuki/  ã                   óˆ   — d dl mZ d dlmZ dedej
                  fd„Zdedej                  fd„Zdej                  defd„Z	y)	é    )Ú
xla_client)ÚGSPMDShardingÚshardingÚreturnc                 óP   — t        j                  «       }|j                  | «       |S ©N)r   Ú
OpShardingÚParseFromString)r   Úprotos     úN/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/extend/sharding.pyÚ%get_op_sharding_from_serialized_protor      s#   € ä
×
Ñ
Ó
!€%Ø×Ñ˜Ô!Ø	€,ó    c                 óR   — t         j                  j                  t        | «      «      S r   )r   ÚHloShardingÚ
from_protor   ©r   s    r   Ú&get_hlo_sharding_from_serialized_protor      s#   € ä	×	Ñ	×	*Ñ	*Ü+¨HÓ5ó
7ð 7r   c                 ó>   — | j                  «       j                  «       S r   )Úto_protoÚSerializeToStringr   s    r   Ú&get_serialized_proto_from_hlo_shardingr   "   s   € à	×	Ñ	Ó	×	.Ñ	.Ó	0Ð0r   N)
Újax._src.libr   Újax._src.sharding_implsr   Úbytesr	   r   r   r   r   © r   r   ú<module>r      s^   ðõ" $Ý BðØðØ"×-Ñ-óð7Øð7Ø"×.Ñ.ó7ð1Ø×$Ñ$ð1Ø).ô1r   