PR #44506 Tensor Parallelism and `mps` device

Tensor Parallelism and `mps` device #44506

3outeille merged 2 commits into huggingface:main from michaelbenayoun:mps_tp

fix: fail when using `mps` with TP

22609a81

michaelbenayoun requested a review from

3outeille 6 days ago

Merge branch 'main' into mps_tp

a266dee2

3outeille approved these changes on 2026-03-09

3outeille enabled auto-merge (squash) 3 days ago

3outeille merged 78aee7e8 into main 3 days ago

michaelbenayoun deleted the mps_tp branch 1 day ago

Reviewers

3outeille

Assignees

No one assigned

Labels

None yet

Milestone

No milestone