Reduction on all distributed system only return the sum, irrespective of the provided parameter. Only works as intended on multi-cpu.

[AbdurRafae]

https://github.com/huggingface/accelerate/blob/a54cd0abd8f1f1f5e0f64ac030427c30760309d8/src/accelerate/utils/operations.py#L442-L462

[sgugger]

Thanks a lot for flagging! Do you want to open a PR to fix this?

[AbdurRafae]

Yeah, sure. I can try that.

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.