jax
5597e903 - [Mosaic:GPU] Disable set_xla_gpu_enable_allocator_spatial_partitioning for nvshmem test case, since nvshmem can't be used within the same allocation.

Commit

44 days ago

[Mosaic:GPU] Disable set_xla_gpu_enable_allocator_spatial_partitioning for nvshmem test case, since nvshmem can't be used within the same allocation. PiperOrigin-RevId: 932410753

References

#38467 - [Mosaic:GPU] Disable set_xla_gpu_enable_allocator_spatial_partitioning for nvshmem test case, since nvshmem can't be used within the same allocation.

Author

PatriosTheGreat

Committer

Google-ML-Automation

Parents

728e4bfe

jax 5597e903 - [Mosaic:GPU] Disable set_xla_gpu_enable_allocator_spatial_partitioning for nvshmem test case, since nvshmem can't be used within the same allocation.

jax
5597e903 - [Mosaic:GPU] Disable set_xla_gpu_enable_allocator_spatial_partitioning for nvshmem test case, since nvshmem can't be used within the same allocation.