Abstract:Recent advancements in large language models (LLMs) boasting billions of parameters have generated a significant demand for efficient deployment in inference workloads. While hardware accelerators for Transformer-based models have been extensively studied, the majority of existing approaches rely on temporal architectures that reuse hardware units for different network layers and operators. However, these methods often encounter challenges in achieving low latency due to considerable memory access overhead. This paper investigates the feasibility and potential of model-specific spatial acceleration for LLM inference on FPGAs. Our approach involves the specialization of distinct hardware units for specific operators or layers, facilitating direct communication between them through a dataflow architecture while minimizing off-chip memory accesses. We introduce a comprehensive analytical model for estimating the performance of a spatial LLM accelerator, taking into account the on-chip compute and memory resources available on an FPGA. This model can be extended to multi-FPGA settings for distributed inference. Through our analysis, we can identify the most effective parallelization and buffering schemes for the accelerator and, crucially, determine the scenarios in which FPGA-based spatial acceleration can outperform its GPU-based counterpart. To enable more productive implementations of an LLM model on FPGAs, we further provide a library of high-level synthesis (HLS) kernels that are composable and reusable. This library will be made available as open-source. To validate the effectiveness of both our analytical model and HLS library, we have implemented BERT and GPT2 on an AMD Xilinx Alveo U280 FPGA device. Experimental results demonstrate our approach can achieve up to 13.4 × speedup when compared to previous FPGA-based accelerators for the BERT model. For GPT generative inference, we attain a 2.2 × speedup compared to DFX, an FPGA overlay, in the prefill stage, while achieving a 1.9 × speedup and a 5.7 × improvement in energy efficiency compared to the NVIDIA A100 GPU in the decode stage.

Exploiting Intel Advanced Matrix Extensions (AMX) for Large Language Model Inference

Efficient LLM Inference with I/O-Aware Partial KV Cache Recomputation

Inference Performance Optimization for Large Language Models on CPUs

Efficient LLM inference solution on Intel GPU

Task Scheduling for Efficient Inference of Large Language Models on Single Moderate GPU Systems

Harnessing Your DRAM and SSD for Sustainable and Accessible LLM Inference with Mixed-Precision and Multi-level Caching

Inference Acceleration for Large Language Models on CPUs

Efficient Arbitrary Precision Acceleration for Large Language Models on GPU Tensor Cores

High-throughput Generative Inference of Large Language Models with a Single GPU

RTiL: Real-Time Inference of Large Language Models on Memory-Constrained GPU Devices

Efficient Deployment of Large Language Model Across Cloud-Device Systems

PowerInfer: Fast Large Language Model Serving with a Consumer-grade GPU

FlashDecoding++: Faster Large Language Model Inference on GPUs

Understanding the Potential of FPGA-Based Spatial Acceleration for Large Language Model Inference

A Hardware Evaluation Framework for Large Language Model Inference

LLM in a flash: Efficient Large Language Model Inference with Limited Memory

MoE-Lightning: High-Throughput MoE Inference on Memory-constrained GPUs

Distributed Inference Performance Optimization for LLMs on CPUs

InstInfer: In-Storage Attention Offloading for Cost-Effective Long-Context LLM Inference