Benjamin Chetioui d09df7c8ab [Mosaic GPU] Add transform inference rules for mgpu.async_{load,store}. ...

PiperOrigin-RevId: 736795784

2025-03-14 02:37:55 -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

[mosaic_gpu/pallas:mgpu] Eradicate wgmma_layout

2025-03-12 10:47:48 -07:00

gpu_test.py

[mosaic_gpu/pallas:mgpu] Eradicate wgmma_layout

2025-03-12 10:47:48 -07:00

gpu_transform_inference_test.py

[Mosaic GPU] Add transform inference rules for mgpu.async_{load,store}.

2025-03-14 02:37:55 -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

[Mosaic GPU] Add CUPTI profiler alongside events-based implementation

2024-12-09 14:31:20 -05:00