jax
02c19267 - Merge pull request #42478 from tanveer-arch:fix-choose-narrow-dtype-large-n

Commit

4 days ago

Merge pull request #42478 from tanveer-arch:fix-choose-narrow-dtype-large-n PiperOrigin-RevId: 1019347064

References

#41702 - [ROCm] Trim wheel RUNPATH: drop per-family, tarball, cross-Python and other-depth entries

#42272 - [ROCm] Build the ROCm plugin with static libc++

#42328 - [ROCm] Use C++20 for hermetic ROCm builds

#42478 - Fix jnp.choose for narrow index dtypes and large choice counts

#42552 - [ROCm] Fix tests that assume exactly 8 devices

#42612 - [ROCm] Build debug-symbol and ASAN ROCm wheels in nightly and continuous CI

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading