jax
c08d31c1 - [Mosaic GPU][NFC] Remove the default arguments of `SmemDescriptor.slice_i64`.

Commit

4 days ago

[Mosaic GPU][NFC] Remove the default arguments of `SmemDescriptor.slice_i64`. All callers pass `mn_idx` and `k_idx` explicitly. PiperOrigin-RevId: 1019098751

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

#42516 - [Mosaic GPU][NFC] Remove the default arguments of `SmemDescriptor.slice_i64`.

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

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

Author

allanrenucci

Committer

Google-ML-Automation

Parents

b82be556

jax c08d31c1 - [Mosaic GPU][NFC] Remove the default arguments of `SmemDescriptor.slice_i64`.

jax
c08d31c1 - [Mosaic GPU][NFC] Remove the default arguments of `SmemDescriptor.slice_i64`.