Ë
    ©sgp  ã                   ó(   — d dl mZ ddlmZ defd„Zy)é    )Ú
DataLoaderé   )Úis_torch_xla_availableÚ
dataloaderc                 óÚ   — t        «       r`dd lmc m} t	        | |j
                  «      sJ d«       ‚dd lmc m} |j                  |j                  «       d«      }|| j                  d<   | S | S )Nr   zPThe dataloader must be a `torch_xla.distributed.parallel_loader.MpDeviceLoader`.)ÚfsdpNÚinput_sharding)r   Ú%torch_xla.distributed.parallel_loaderÚdistributedÚparallel_loaderÚ
isinstanceÚMpDeviceLoaderÚtorch_xla.distributed.spmdÚspmdÚShardingSpecÚget_global_meshÚ_parallel_loader_kwargs)r   ÚplÚxsÚsharding_specs       úP/var/www/html/venv/lib/python3.12/site-packages/transformers/integrations/tpu.pyÚtpu_spmd_dataloaderr      ss   € ÜÔß:Ð:äØ˜×)Ñ)ô
ð 	^à]ó	^ð 
÷ 	0Ð/àŸ™¨×(:Ñ(:Ó(<¸nÓMˆØ?Lˆ
×*Ñ*Ð+;Ñ<ØÐàÐó    N)Útorch.utils.datar   Úutilsr   r   © r   r   ú<module>r      s   ðõ (å *ð Jô r   