jax
21efe9a6 - [pallas:sc] `pltpu.sync_copy` now accepts `add=`

Commit

3 days ago

[pallas:sc] `pltpu.sync_copy` now accepts `add=` Previously, the only way to do a DMA update was to use `pltpu.async_copy`. PiperOrigin-RevId: 866439264

References

#34881 - [pallas:sc] `pltpu.sync_copy` now accepts `add=`

Author

superbobry

Committer

Google-ML-Automation

Parents

89fc1a93

jax 21efe9a6 - [pallas:sc] `pltpu.sync_copy` now accepts `add=`

jax
21efe9a6 - [pallas:sc] `pltpu.sync_copy` now accepts `add=`