Closed fhossein-quic closed 1 month ago
tt.get_program_id x
is target dependent, so we didn't handle these ops in triton-linalg
. It can be convert to gpu.block_id x
under the GPU or omp dialect for CPU.auxilary
can be directly lower to llvm dialect
as it just bind llvm.ptr to memref desc, while linalgext
can be lower to vector or other target dependent dialect
. you may refer to the codegen dialect overview of llvm community
I'm trying to use triton-linalg tool to lower ttir to linalg, but realized that after conversion,
tt.get_program_id
andtt.get_num_programs
still remain in the tt dialect and are not lowered to linalg. The add_kernel example shared in the repo's README shows this issue as well:I'd also like to know how ops in
Auxilary
andLinalgExt
Dialect can lowered to LLVM IR? What dialects they can be transformed to?CC- @sethbrin and @zhangqirui