Adam Paszke bb96226dd8 [Mosaic GPU] Add support for small RHS tile sizes in WGMMA ...

This is useful for more fine-grained autotuning and can help avoid
wave quantization effects.

PiperOrigin-RevId: 732105219

2025-02-28 05:41:30 -08: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 an abstraction to obtain a slice of dynamic shared memory when using waprgroup semantics.

2025-02-28 04:38:25 -08: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] Add support for small RHS tile sizes in WGMMA

2025-02-28 05:41:30 -08: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