Loading [a11y]/accessibility-menu.js
A Code Generator for High-Performance Tensor Contractions on GPUs | IEEE Conference Publication | IEEE Xplore

A Code Generator for High-Performance Tensor Contractions on GPUs


Abstract:

Tensor contractions are higher dimensional generalizations of matrix-matrix multiplication. They form the compute-intensive core of many applications in computational sci...Show More

Abstract:

Tensor contractions are higher dimensional generalizations of matrix-matrix multiplication. They form the compute-intensive core of many applications in computational science and data science. In this paper, we describe a high-performance GPU code generator for arbitrary tensor contractions. It exploits domain-specific properties about data reuse in tensor contractions to devise an effective code generation schema, coupled with an effective model-driven search, to determine parameters for mapping of computation to threads and staging of data through the GPU memory hierarchy. Experimental evaluation using a set of tensor contraction benchmarks demonstrates performance improvement and/or significantly reduced code generation time over other state-of-the-art tensor contraction libraries and code generators.
Date of Conference: 16-20 February 2019
Date Added to IEEE Xplore: 07 March 2019
ISBN Information:
Conference Location: Washington, DC, USA

Contact IEEE to Subscribe

References

References is not available for this document.