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.
翻译:为解决张量运算缺乏通用标准接口的问题,我们提出了张量代数处理原语(TAPP),这是一个基于C语言的接口,旨在将应用层与硬件特定实现解耦。我们提供了张量缩并的数学表述及参考实现,以确保正确性并促进优化内核的验证。TAPP通过凝聚学术界与工业界共识开发而成,致力于实现性能可移植性并解决依赖性问题。该标准的可行性已通过其与TBLIS、cuTENSOR库以及DIRAC量子化学软件包的成功集成得到验证。