Benjamin Chetioui d3bf243342 [Mosaic GPU] Add layout inference for splat arith.ConstantOps and vector.SplatOps. ...

PiperOrigin-RevId: 716224880

2025-01-16 07:18:35 -08:00

..

BUILD

Rename test configs to include GPU variants more consistently.

2025-01-14 11:55:45 -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

[MosaicGPU] Move gpu_address_space_to_nvptx inside utils.py and use it.

2025-01-16 06:41:51 -08:00

gpu_layout_inference_test.py

[Mosaic GPU] Add layout inference for splat arith.ConstantOps and vector.SplatOps.

2025-01-16 07:18:35 -08:00

gpu_test.py

[MosaicGPU] Remove the single_thread context from top-level dialect code.

2025-01-16 06:59:32 -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