ó
    eT–hr  ã                   ó*   • S SK Jr  SSKJr  S\4S jrg)é    )Ú
DataLoaderé   )Úis_torch_xla_availableÚ
dataloaderc                 óê   • [        5       (       ac  SS KJs  Jn  [	        XR
                  5      (       d   S5       eSS KJs  Jn  UR                  UR                  5       S5      nX0R                  S'   U $ U $ )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       ÚU/var/www/auris/envauris/lib/python3.13/site-packages/transformers/integrations/tpu.pyÚtpu_spmd_dataloaderr      sn   € Ü×Ñß:Ð:ä˜*×&7Ñ&7×8Ñ8ð 	
Ø^ó	
Ð8÷ 	0Ð/àŸ™¨×(:Ñ(:Ó(<¸nÓMˆØ?L×*Ñ*Ð+;Ñ<ØÐàÐó    N)Útorch.utils.datar   Úutilsr   r   © r   r   Ú<module>r      s   ðõ (å *ð Jõ r   