Brown Shield

Scientific Computing Group Report - Detail view

Identifier: 2009-40

Author(s): A. Kloeckner, N. Pinto, Y. Lee, B. Catanzaro, P. Ivanov, and A. Fasih

Title: PyCUDA: GPU Run-Time Code Generation for High-Performance Computing

Page count: 19 pp.

Date: Nov. 18, 2009

Abstract:

High-performance scientific computing has recently seen a surge of interest in heterogeneous systems, with an emphasis on modern Graphics Processing Units (GPUs). These devices offer tremendous potential for performance and efficiency in important large-scale applications of computational science. However, exploiting this potential can be challenging, as one must adapt to the specialized and rapidly evolving computing environment currently exhibited by GPUs. One way of addressing this challenge is to embrace better techniques and develop tools tailored to their needs. This article presents one simple technique, GPU run-time code generation (RTCG), and PyCUDA, an open-source toolkit that supports this technique. In introducing PyCUDA, this article proposes the combination of a dynamic, high-level scripting lan- guage with the massive performance of a GPU as a compelling two-tiered computing platform, potentially offering significant performance and productivity advantages over conventional single-tier, static systems. It is further observed that, compared to competing techniques, the effort required to create codes using run-time code generation with PyCUDA grows more gently in response to growing needs. The concept of RTCG is simple and easily implemented using existing, robust tools. Nonetheless it is powerful enough to support (and encourage) the creation of custom application-specific tools by its users. The premise of the paper is illustrated by a wide range of examples where the technique has been applied with considerable success.

BibTeX:

@techreport{brown_sc_2009_40,
  title = "{PyCUDA: GPU Run-Time Code Generation for High-Performance Computing}",
  author = "A. Kloeckner, N. Pinto, Y. Lee, B. Catanzaro, P. Ivanov, and A. Fasih",
  institution = "Scientific Computing Group, Brown University",
  number = "2009-40",
  address = "Providence, RI, USA",
  year = 2009,
  month = nov,
}

Download: