Gleb Pobudzey 54691b125a [Mosaic GPU] Support reads/writes from SMEM to WGMMARowFragLayout arrays. ...

PiperOrigin-RevId: 738121106

2025-03-18 13:23:07 -07:00

..

BUILD

[Mosaic GPU] Introduce an initial transform inference pass.

2025-03-13 23:22:59 -07:00

flash_attention_test.py

[Mosaic GPU] Only run tests requiring sm90a on Hopper

2024-11-11 12:02:48 -05:00

gpu_dialect_test.py

[Pallas/Mosaic GPU] Add lowerings/layout inference for all the necessary conversion ops when using Warpgroup semantics.

2025-03-10 02:14:39 -07:00

gpu_layout_inference_test.py

[pallas:mosaic_gpu] jnp.reduce_sum now works for >1D arrays

2025-03-17 05:32:07 -07:00

gpu_test.py

[Mosaic GPU] Support reads/writes from SMEM to WGMMARowFragLayout arrays.

2025-03-18 13:23:07 -07:00

gpu_transform_inference_test.py

[Mosaic GPU] Add transform inference rule for mgpu.slice_smem.

2025-03-18 04:53:54 -07:00

matmul_test.py

[Mosaic GPU] Only run tests requiring sm90a on Hopper

2024-11-11 12:02:48 -05:00

profiler_cupti_test.py

Removed trivial docstrings from JAX tests

2025-03-17 07:49:37 -07:00