pytorch

7806f748 - Forward fix for DataLoader to accept custom Sharding DataPipe (#97287)

Forward fix for DataLoader to accept custom Sharding DataPipe (#97287) Fixes #96975 Changes: - Make sure custom ShardingDataPipe with `apply_sharding` can be used by `DataLoader` - Allow the `apply_sharding` function without the last argument of `sharding_group` - Make `DataLoader` not relying on `sharding_group` Pull Request resolved: https://github.com/pytorch/pytorch/pull/97287 Approved by: https://github.com/NivekT