Enabling Long FFT Convolutions on Memory-Constrained FPGAs via Chunking

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Peter, Gupta, Neelesh, Prasanna, Viktor
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914244514021376
author Wang, Peter
Gupta, Neelesh
Prasanna, Viktor
author_facet Wang, Peter
Gupta, Neelesh
Prasanna, Viktor
contents The need for long-context reasoning has led to alternative neural network architectures besides Transformers and self-attention, a popular model being Hyena, which employs causal 1D-convolutions implemented with FFTs. Long convolutions enable efficient global context mixing, but requirements for intermediate results exceed the 2-3 MB Block RAM capacity of FPGAs. We present a chunked FFT convolution approach enabling 450K length sequence by 450K length filter convolutions on an Alveo U200 FPGA with 2.8 MB BRAM through chunking and overlap-add reconstruction. We find that throughput scales proportionally with chunk size while degrading minimally by 7% for our longest sequences, demonstrating that careful memory management enables deployment of long-context primitives on edge FPGAs without sacrificing performance.
format Preprint
id arxiv_https___arxiv_org_abs_2601_06065
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Enabling Long FFT Convolutions on Memory-Constrained FPGAs via Chunking
Wang, Peter
Gupta, Neelesh
Prasanna, Viktor
Machine Learning
Hardware Architecture
The need for long-context reasoning has led to alternative neural network architectures besides Transformers and self-attention, a popular model being Hyena, which employs causal 1D-convolutions implemented with FFTs. Long convolutions enable efficient global context mixing, but requirements for intermediate results exceed the 2-3 MB Block RAM capacity of FPGAs. We present a chunked FFT convolution approach enabling 450K length sequence by 450K length filter convolutions on an Alveo U200 FPGA with 2.8 MB BRAM through chunking and overlap-add reconstruction. We find that throughput scales proportionally with chunk size while degrading minimally by 7% for our longest sequences, demonstrating that careful memory management enables deployment of long-context primitives on edge FPGAs without sacrificing performance.
title Enabling Long FFT Convolutions on Memory-Constrained FPGAs via Chunking
topic Machine Learning
Hardware Architecture
url https://arxiv.org/abs/2601.06065