DeepSpeed

Fix broadcast deadlock for incomplete batches in data sample for data analysis

#5117

Merged