[Context Parallel] fix: support PyTorch < 2.6 in distributed gather#13075

Open

akshatvishu wants to merge 3 commits intohuggingface:mainfrom

akshatvishu:fix-torch-accelerator-compatibility

Commits on Feb 3, 2026

[Context Parallel] fix: support PyTorch < 2.6 in distributed gather
akshatvishu
committed

Commits on Feb 4, 2026

[Context Parallel] fix: use get_device() and add fail-fast check
akshatvishu
committed
fix: use torch.device with get_device and simplify error message
akshatvishu
committed