jax
a320c518 - Merge pull request #41856 from stroxler:fix-noncontiguous-overload

Commit

9 days ago

Merge pull request #41856 from stroxler:fix-noncontiguous-overload PiperOrigin-RevId: 1004026981

References

#41703 - [ROCm] Build and test the ROCm wheels in the rocmless container with pip-installed TheRock

#41856 - [typing] Avoid non-contiguous overloads in splash_attention_kernel.py

#41881 - [ROCm] Enable remote compilation for jax in rocm rbe

#42141 - [ROCm] Run wheel packaging locally with remote compilation

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading