xla
462cb0f4 - Introduce `torch_xla.runtime.use_spmd()` (#5474)

Commit

2 years ago

Introduce `torch_xla.runtime.use_spmd()` (#5474) Introduce torch_xla.runtime.use_spmd() and torch_xla.runtime.is_spmd()

References

Author

yeounoh

Parents