A note on auto-tuning GEMM for GPUs

Yinan Li, Jack Dongarra, Stanimire Tomov

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

120 Scopus citations

Abstract

The development of high performance dense linear algebra (DLA) critically depends on highly optimized BLAS, and especially on the matrix multiplication routine (GEMM). This is especially true for Graphics Processing Units (GPUs), as evidenced by recently published results on DLA for GPUs that rely on highly optimized GEMM. However, the current best GEMM performance, e.g. of up to 375 GFlop/s in single precision and of up to 75 GFlop/s in double precision arithmetic on NVIDIA's GTX 280, is difficult to achieve. The development involves extensive GPU knowledge and even backward engineering to understand some undocumented insides about the architecture that have been of key importance in the development. In this paper, we describe some GPU GEMM auto-tuning optimization techniques that allow us to keep up with changing hardware by rapidly reusing, rather than reinventing, the existing ideas. Auto-tuning, as we show in this paper, is a very practical solution where in addition to getting an easy portability, we can often get substantial speedups even on current GPUs (e.g. up to 27% in certain cases for both single and double precision GEMMs on the GTX 280).

Original languageEnglish
Title of host publicationComputational Science - ICCS 2009 - 9th International Conference, Proceedings
Pages884-892
Number of pages9
EditionPART 1
DOIs
StatePublished - 2009
Event9th International Conference on Computational Science, ICCS 2009 - Baton Rouge, LA, United States
Duration: May 25 2009May 27 2009

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
NumberPART 1
Volume5544 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference9th International Conference on Computational Science, ICCS 2009
Country/TerritoryUnited States
CityBaton Rouge, LA
Period05/25/0905/27/09

Keywords

  • Auto-tuning
  • Dense linear algebra
  • GPUs
  • Matrix multiply

Fingerprint

Dive into the research topics of 'A note on auto-tuning GEMM for GPUs'. Together they form a unique fingerprint.

Cite this