Generating Performance Portable Code using Rewrite Rules: From High-level Functional Expressions to High-Performance OpenCL Code
The University of Edinburgh, UK
The University of Edinburgh, 2015
@article{steuwer2015generating,
title={Generating Performance Portable Code using Rewrite Rules},
author={Steuwer, Michel and Fensch, Christian and Lindley, Sam and Dubach, Christophe},
year={2015}
}
Computing systems have become increasingly complex with the emergence of heterogeneous hardware combining multicore CPUs and GPUs. These parallel systems exhibit tremendous computational power at the cost of increased programming effort. This results in a tension between performance and code portability. Typically, code is either tuned in an low-level imperative language using hardware-specific optimizations to achieve maximum performance or is written in a high-level, possibly functional, language to achieve portability at the expense of performance. We propose a novel approach aiming to combine high-level programming, code portability, and high-performance. Starting from a high-level functional expression we apply a simple set of rewrite rules to transform it into a low-level functional representation close to the OpenCL programming model from which OpenCL code is generated. Our rewrite rules define a space of possible implementations which we automatically explore to generate hardware-specific OpenCL implementations. We formalise the system with a core dependently-typed lambda-calculus along with a denotational semantics which we use to prove the correctness of the rewrite rules. We test our design by describing a subset of the OpenCL programming model in a functional style and by implementing a compiler which generates high performance imperative OpenCL code. Our experiments show that we can automatically derive high-performance hardware-specific implementations from simple functional high-level algorithmic expressions. The performance of the generated OpenCL code is on a par with highly tuned implementations for multicore CPUs and GPUs written by experts.
March 8, 2015 by hgpu