Fix device assignment in `get_device_name` for distributed training#3303

Merged

tomaarsen merged 2 commits intoUKPLab:masterfrom

uminaty:fix-get-device-name

Apr 3, 2025

Commits on Apr 2, 2025

Fix get_device_name for distributed setup
uminaty
committed

Commits on Apr 3, 2025

Clean up local_rank assignment logic in get_device_name

uminaty
and
tomaarsen
authored