Dimitar (Mitko) Asenov d2bf034c47 [Mosaic GPU] Test the wgmma_op lowering when a is in registers. ...

I had to add support for wgmma layout in vector_load. Not sure if this is useful outside the test.

PiperOrigin-RevId: 735384104

2025-03-10 08:25:43 -07:00

..

BUILD

[Mosaic GPU] Add support for small RHS tile sizes in WGMMA

2025-02-28 05:41:30 -08: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] Make sure to relayout FAs when their layouts mismatch in MGPU lowering

2025-02-26 04:03:57 -08:00

gpu_test.py

[Mosaic GPU] Test the wgmma_op lowering when a is in registers.

2025-03-10 08:25:43 -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