jax
9e11f1c0 - [Pallas:TPU] Add tests for DMAs with regular semaphores

Commit

2 days ago

[Pallas:TPU] Add tests for DMAs with regular semaphores PiperOrigin-RevId: 875628932

References

#26604 - [ROCm][pallas:triton] Fix atomic min/max lowering for uint and float types

#34702 - [ROCm] Add script for running bazel tests on ROCm

#34722 - [ROCm] ToT ROCm Unit Test Skips

#35364 - [Pallas:TPU] Add tests for DMAs with regular semaphores

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading