mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 20:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/tests/mosaic
History
Sergei Lebedev a373e37be2 Fixed mgpu.FragmentedArray.reduce_sum for integer types 
The implementation previously assumed the type is floating and used addf.

PiperOrigin-RevId: 678718871
2024-09-25 08:50:24 -07:00
..
BUILD
Enable more H100 tests in CI.
2024-09-25 07:37:48 -07:00
flash_attention_test.py
Update minimum Python version to 3.10.
2024-06-26 13:47:14 -04:00
gpu_test.py
Fixed mgpu.FragmentedArray.reduce_sum for integer types
2024-09-25 08:50:24 -07:00
matmul_test.py
[Mosaic GPU] Add support for grid tiling to improve L2 cache utilization
2024-08-14 02:17:55 -07:00