jax
1c817be6
- [Pallas:MGPU] Expose cluster launch controls in Pallas, allowing an SM to dynamically query the next unit of work to schedule.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
103 days ago
[Pallas:MGPU] Expose cluster launch controls in Pallas, allowing an SM to dynamically query the next unit of work to schedule. PiperOrigin-RevId: 811450394
References
#31381 - Remove old ROCm build code
#32109 - [Pallas:MGPU] Expose cluster launch controls in Pallas, allowing an SM to dynamically query the next unit of work to schedule.
#32115 - Relax version requirements for ROCm Jax Plugin wheels
#33157 - Resolve undefined behavior in bitshift unit test
#33186 - Make nvidia version data optional for ROCm builds
#579 - Create rocm-test-requirements.txt
#580 - GESVDJ support for ROCm GPUs in JAX
#581 - Fix/pallas tests shared memory
#584 - Use plain bazel to test jax, use hermetic rocm dependency
#585 - update a test for checking zero ROCm GPU event
#34135 - [ROCm] update to test if there are GPU events when doing profiling on…
#587 - Mark ROCm support for gtsv2 as stable
Author
justinjfu
Committer
Google-ML-Automation
Parents
262b80c9
Loading