pytorch

03ed8cbf - Workaround for bug in DistributedDataParallel (#46385)

Workaround for bug in DistributedDataParallel (#46385) Fix the DistributedDataParallelSingleProcessTest to work around a limitation in DistributedDataParallel where the batch_size needs to evenly divide by the number of GPUs used See #46175