PR #3952 Fix DDP initialization with XLA backend

Fix DDP initialization with XLA backend #3952

will-cromar merged 5 commits into master from pjrt-ddp-init

Fix DDP initialization with XLA backend

b5c27f85

Add test case

e12e4608

will-cromar added runtime

will-cromar added ddp

will-cromar requested a review from

alanwaketan 3 years ago

will-cromar requested a review from

JackCaoG 3 years ago

alanwaketan commented on 2022-08-30

Formatting and review comments.

ef539a8d

alanwaketan approved these changes on 2022-08-30

JackCaoG approved these changes on 2022-08-31

JackCaoG commented on 2022-09-01

yeounoh commented on 2022-09-01

Implement `xm.collective_permute`

baa1aedb

Clarify implementation for `collective_permute`

134fcdd6

alanwaketan commented on 2022-09-03

will-cromar requested a review from

alanwaketan 3 years ago

will-cromar requested a review from

JackCaoG 3 years ago

alanwaketan approved these changes on 2022-09-06

will-cromar merged 6e42e7cb into master 3 years ago

Reviewers

alanwaketan

JackCaoG

yeounoh

Assignees

No one assigned

Labels

runtime ddp

Milestone

No milestone