Publication: AXI4MLIR: User-Driven automatic host code generation for custom AXI-Based accelerators
Authors
Bohm Agostini, Nicolas ; Haris, Jude ; Gibson, Perry ; Jayaweera, Malith ; Rubin, Norm ; Tumeo, Antonino ; Abellán, José L. ; Cano, José ; Kaeli, David
item.page.secondaryauthor
item.page.director
Publisher
publication.page.editor
publication.page.department
DOI
https://doi.org/10.48550/arXiv.2312.14821
item.page.type
info:eu-repo/semantics/article
Description
This document is a PrePrint . You can find it also in arXiv.org, with DOI: https://doi.org/10.48550/arXiv.2312.14821
Abstract
This paper addresses the need for automatic and efficient generation of host driver code for arbitrary custom AXI-based accelerators targeting linear algebra algorithms, an important workload in various applications, including machine learning and scientific computing. While existing tools have focused on automating accelerator prototyping, little attention has been paid to the host-accelerator interaction. This paper introduces AXI4MLIR, an extension of the MLIR compiler framework designed to facilitate the automated generation of host-accelerator driver code. With new MLIR attributes and transformations, AXI4MLIR empowers users to specify accelerator features (including their instructions) and communication patterns and exploit the host memory hierarchy. We demonstrate AXI4MLIR's versatility across different types of accelerators and problems, showcasing significant CPU cache reference reductions (up to 56%) and up to a 1.65x speedup compared to manually optimized driver code implementations. AXI4MLIR implementation is open-source and available at: t: https://github.com/AXI4MLIR/axi4mlir
publication.page.subject
Citation
item.page.embargo
Collections
Ir a Estadísticas
Este ítem está sujeto a una licencia Creative Commons. http://creativecommons.org/licenses/by-nc-nd/4.0/