jax
9f654f56 - [Mosaic] Add target core type parameter to tpu.enqueue_dma

Commit

4 days ago

[Mosaic] Add target core type parameter to tpu.enqueue_dma This is similar to how `core_type` is used for tpu.semaphore_signal PiperOrigin-RevId: 866722791

References

#34426 - [Mosaic] Add target core type parameter to tpu.enqueue_dma

Author

vsytch

Committer

Google-ML-Automation

Parents

ba8ebca5

jax 9f654f56 - [Mosaic] Add target core type parameter to tpu.enqueue_dma

jax
9f654f56 - [Mosaic] Add target core type parameter to tpu.enqueue_dma