arXiv · 2410.00825
Developing a BLAS library for the AMD AI Engine
Abstract
Spatial (dataflow) computer architectures can mitigate the control and performance overhead of classical von Neumann architectures such as traditional CPUs. Driven by the popularity of Machine Learning (ML) workloads, spatial devices are being marketed as ML inference accelerators. Despite providing a rich software ecosystem for ML practitioners, their adoption in other scientific domains is hindered by the steep learning curve and lack of reusable software, which makes them inaccessible to non-experts. We present our ongoing project AIEBLAS, an open-source, expandable implementation of Basic Linear Algebra Routines (BLAS) for the AMD AI Engine. Numerical routines are designed to be easily reusable, customized, and composed in dataflow programs, leveraging the characteristics of the targeted device without requiring the user to deeply understand the underlying hardware and programming model.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Tristan Laan, Tiziano De Matteis. 2024-10-01. Developing a BLAS library for the AMD AI Engine. https://arxiv.org/abs/2410.00825
Cite the original work for its findings. Save a collection to share your selection of sources.