vllm
[TPU] Support collective communications in XLA devices
#6813

Merged

[TPU] Support collective communications in XLA devices #6813

WoosukKwon merged 6 commits into main from add-xla-comm

[TPU] Support collective communications in XLA devices

af3a259f

WoosukKwon added tpu

WoosukKwon added ready

WoosukKwon requested a review from

youkaichao 1 year ago

Use current_platform

0f2abea4

youkaichao commented on 2024-07-26

is_xla -> is_tpu

8ebea7ea

WoosukKwon requested a review from

youkaichao 1 year ago

youkaichao commented on 2024-07-26

Define TPU communicator

782b1828

youkaichao commented on 2024-07-26

Fix

80872273

Address comments

f04e1792

youkaichao commented on 2024-07-26

youkaichao approved these changes on 2024-07-26

WoosukKwon enabled auto-merge (squash) 1 year ago

WoosukKwon merged d09b94ca into main 1 year ago

WoosukKwon deleted the add-xla-comm branch 1 year ago

Reviewers

youkaichao

Assignees

No one assigned

Labels

tpu ready

Milestone

No milestone