Tensor Algebra Processing Primitives (TAPP): Towards a Standard for Tensor Operations

Read full text
Authors: Jan Brandejs, Niklas Hörnblad, Edward F. Valeev, Alexander Heinecke, Jeff Hammond, Devin Matthews, Paolo Bientinesi
Cited by: 1

Abstract

To address the absence of a universal standard interface for tensor operations, we introduce the Tensor Algebra Processing Primitives (TAPP), a C-based interface designed to decouple the application layer from hardware-specific implementations. We provide a mathematical formulation of tensor contractions and a reference implementation to ensure correctness and facilitate the validation of optimized kernels. Developed through community consensus involving academic and industrial stakeholders, TAPP aims to enable performance portability and resolving dependency challenges. The viability of the standard is demonstrated through successful integrations with the TBLIS and cuTENSOR libraries, as well as the DIRAC quantum chemistry package.

Fields of Study & Subjects

Parallel Computing and Optimization Techniques Scientific Computing and Data Management Machine Learning in Materials Science

Publication details

Journal / source
arXiv (Cornell University)
Publication date
12 January 2026
Document type
Preprint