Custom batch sampler and DDP

[Jserax]

Hello, when I use a custom batch sampler with DDP, do I need to modify it with split dataset for 2 GPUs, similar to DistributedSampler from PyTorch, or will Lightning handle everything automatically?