accelerate
15163792 - Better error when device mismatches when calling gather() on CUDA (#2180)

Commit
2 years ago
Better error when device mismatches when calling gather() on CUDA (#2180) * Better err * Update src/accelerate/utils/operations.py Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com> --------- Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
Author
Parents
Loading