xla
Enable PJRT C API Client and other changes for Neuron
#5428

Merged

Enable PJRT C API Client and other changes for Neuron #5428

JackCaoG merged 9 commits into pytorch:master from aws-kingrj:pjrt_neuron

Enable PJRT C API Client and other changes for Neuron

32537ea8

JackCaoG requested a review from

will-cromar 2 years ago

JackCaoG added runtime

keeping quotes consistent

20c74389

fixing device type call

496d7b2a

will-cromar commented on 2023-08-09

refactoring neuron initialization with spawn

50cc08b3

updating replication setting only for torchrun

e3604f07

will-cromar commented on 2023-08-10

removing set replication in xla backed was added to rendezvous handler

bcc68aba

Merge branch 'master' into pjrt_neuron

8e3c342d

removing workaround for world_size/master_port for neuron

4b1a3fc5

fixing linter issues

ee23f84b

will-cromar approved these changes on 2023-08-15

JackCaoG merged daed1d0a into master 2 years ago

Reviewers

will-cromar

Assignees

No one assigned

Labels

runtime

Milestone

No milestone