` `init_method` (#5438)

Commit

2 years ago

[PJRT] Support `torchrun` with `pjrt://` `init_method` (#5438) * Support torchrun with `pjrt://` `init_method` * move import * fix error * Fix NameError * Fix path * Remove from TPU CI

References

#5438 - [PJRT] Support `torchrun` with `pjrt://` `init_method`

Author

will-cromar

Parents

5b88b5f8

xla 9db8fac4 - [PJRT] Support `torchrun` with `pjrt://` `init_method` (#5438)

xla
9db8fac4 - [PJRT] Support `torchrun` with `pjrt://` `init_method` (#5438)