[GPU] Add basic GPU support and example

buddy-compiler / buddy-mlir

An MLIR-based compiler framework bridges DSLs (domain-specific languages) to DSAs (domain-specific architectures).

https://buddy-compiler.github.io/

Apache License 2.0

485 stars 159 forks source link

Open matrix72c opened 1 week ago

matrix72c commented 1 week ago

Basic GPU support for lowering linalg.matmul to GPU. Integrated from #290

Create Buddy GPU Dialact, and add three ops: hoist_static_alloc, unroll_vectors_gpu_mma_sync, vector_to_mma_conversion for transform dialact inside GPU dialact.
Extend MLIRGPU in Conversion, adding two passes: ConvertMemcpyToGPUPass and LegalizeShmemOutliningPass.
Add tests for both ops and passes.
Add a detailed example in examples/BuddyGPU.