§
    ñ§gp  ã                   ó*   — d dl mZ ddlmZ defd„ZdS )é    )Ú
DataLoaderé   )Úis_torch_xla_availableÚ
dataloaderc                 óì   — t          ¦   «         redd lmc m} t	          | |j        ¦  «        s
J d¦   «         ‚dd lmc m} |                     | 	                    ¦   «         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       úY/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/integrations/tpu.pyÚtpu_spmd_dataloaderr      s·   € ÝÑÔð Ø:Ð:Ð:Ð:Ð:Ð:Ð:Ð:Ð:åØ˜Ô)ñ
ô 
ð 	^ð 	^à]ñ	^ô 	^ð 	^ð 	0Ð/Ð/Ð/Ð/Ð/Ð/Ð/Ð/àŸš¨×(:Ò(:Ñ(<Ô(<¸nÑMÔMˆØ?Lˆ
Ô*Ð+;Ñ<ØÐàÐó    N)Útorch.utils.datar   Úutilsr   r   © r   r   ú<module>r      sP   ðð (Ð 'Ð 'Ð 'Ð 'Ð 'à *Ð *Ð *Ð *Ð *Ð *ð Jð ð ð ð ð ð r   