xla
daed1d0a - Enable PJRT C API Client and other changes for Neuron (#5428)

Commit
2 years ago
Enable PJRT C API Client and other changes for Neuron (#5428) * Enable PJRT C API Client and other changes for Neuron * keeping quotes consistent * fixing device type call * refactoring neuron initialization with spawn * updating replication setting only for torchrun * removing set replication in xla backed was added to rendezvous handler * removing workaround for world_size/master_port for neuron * fixing linter issues
Author
Parents
Loading