jax
d8bbe601 - [Pallas:MGPU] Expose the leader_tracked in copy_gmem_to_smem

Commit

228 days ago

[Pallas:MGPU] Expose the leader_tracked in copy_gmem_to_smem This will allow us to issue replicated copies with .cta_group::2 PiperOrigin-RevId: 881903544

References

#35721 - [Pallas:MGPU] Expose the leader_tracked in copy_gmem_to_smem

Author

apaszke

Committer

Google-ML-Automation

Parents

422cf9d2

jax d8bbe601 - [Pallas:MGPU] Expose the leader_tracked in copy_gmem_to_smem

jax
d8bbe601 - [Pallas:MGPU] Expose the leader_tracked in copy_gmem_to_smem