jax
2da97136 - [Mosaic GPU] Add `nvshmemx_cumodule_finalize` support.

Commit

58 days ago

[Mosaic GPU] Add `nvshmemx_cumodule_finalize` support. PiperOrigin-RevId: 847665237

References

Author

allanrenucci

Committer

Google-ML-Automation

Parents