The Community for Technology Leaders
2015 44th International Conference on Parallel Processing (ICPP) (2015)
Beijing, China
Sept. 1, 2015 to Sept. 4, 2015
ISSN: 0190-3918
ISBN: 978-1-4673-7587-0
pp: 969-978
ABSTRACT
Many scientific and numerical applications, including quantum chemistry modeling and fluid dynamics simulation, require tensor product and tensor contraction evaluation. Tensor computations are characterized by arrays with numerous dimensions, inherent parallelism, moderate data reuse and many degrees of freedom in the order in which to perform the computation. The best-performing implementation is heavily dependent on the tensor dimensionality and the target architecture. In this paper, we map tensor computations to GPUs, starting with a high-level tensor input language and producing efficient CUDA code as output. Our approach is to combine tensor-specific mathematical transformations with a GPU decision algorithm, machine learning and auto tuning of a large parameter space. Generated code shows significant performance gains over sequential and Open MP parallel code, and a comparison with Open ACC shows the importance of auto tuning and other optimizations in our framework for achieving efficient results.
INDEX TERMS
Tensile stress, Graphics processing units, Computer architecture, DSL, Optimization, Indexes, Parallel processing
CITATION

T. Nelson et al., "Generating Efficient Tensor Contractions for GPUs," 2015 44th International Conference on Parallel Processing (ICPP), Beijing, China, 2015, pp. 969-978.
doi:10.1109/ICPP.2015.106
97 ms
(Ver 3.3 (11022016))