Accelerating GenAI Workloads by Enabling RISC-V Microkernel Support in IREE
Abstract
IREE supports RISC-V microkernel with MLIR linalg dialect optimizations, improving performance for the Llama-3.2-1B-Instruct model compared to upstream IREE and Llama.cpp.
This project enables RISC-V microkernel support in IREE, an MLIR-based machine learning compiler and runtime. The approach begins by enabling the lowering of MLIR linalg dialect contraction ops to linalg.mmt4d op for the RISC-V64 target within the IREE pass pipeline, followed by the development of optimized microkernels for RISC-V. The performance gains are compared with upstream IREE and Llama.cpp for the Llama-3.2-1B-Instruct model.
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper