PR #3609 Adding support for torchrun in xla backend

Adding support for torchrun in xla backend #3609

JackCaoG merged 10 commits into pytorch:master from amithrm:xrt_init

amithrm force pushed from bc05ff87 to 548dd397 3 years ago

JackCaoG requested a review from

will-cromar 3 years ago

will-cromar requested changes on 2022-11-02

JackCaoG commented on 2022-11-02

amithrm force pushed from 548dd397 to 1ecbee88 2 years ago

will-cromar commented on 2022-12-05

amithrm force pushed from 1ecbee88 to eb09f7db 2 years ago

amithrm force pushed from eb09f7db to cdea4ac4 2 years ago

amithrm force pushed from cdea4ac4 to f263be68 2 years ago

amithrm force pushed from f263be68 to e71414a4 2 years ago

Adding support for torchrun in xla backend

87717252

Adding support for custom specification of devices and associated tests

997356b2

Revert "Adding support for custom specification of devices and associ…

8f847162

Adding support for custom specification of devices and associated tests

33360307

Updated xrt initialization to work with TPU based flow

e0c4ff9e

Cleanup xrt initialization to work with torchrun/slurm

6833150f

Adding tests to CI pipeline

d0fbbd8f

Fix Lint/CI issues

b7c02c95

Fixing Lint Error

1d2be473

Lint changes

262813e1

amithrm force pushed from e71414a4 to 262813e1 2 years ago

will-cromar approved these changes on 2022-12-13

will-cromar requested a review from

JackCaoG 2 years ago

JackCaoG approved these changes on 2022-12-13

JackCaoG merged 334d1d66 into master 2 years ago

Reviewers

JackCaoG

will-cromar

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla Adding support for torchrun in xla backend #3609 Merged

Adding support for torchrun in xla backend #3609

xla
Adding support for torchrun in xla backend
#3609

Merged