Heterogeneous Programming and Optimization of Gyrokinetic Toroidal Code Using Directives

  • Wenlu Zhang
  • , Wayne Joubert
  • , Peng Wang
  • , Bei Wang
  • , William Tang
  • , Matthew Niemerg
  • , Lei Shi
  • , Sam Taimourzadeh
  • , Jian Bao
  • , Zhihong Lin

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

The latest production version of the fusion particle simulation code, Gyrokinetic Toroidal Code (GTC), has been ported to and optimized for the next generation exascale GPU supercomputing platform. Heterogeneous programming using directives has been utilized to balance the continuously implemented physical capabilities and rapidly evolving software/hardware systems. The original code has been refactored to a set of unified functions/calls to enable the acceleration for all the species of particles. Extensive GPU optimization has been performed on GTC to boost the performance of the particle push and shift operations. In order to identify the hotspots, the code was the first benchmarked on up to 8000 nodes of the Titan supercomputer, which shows about 2–3 times overall speedup comparing NVidia M2050 GPUs to Intel Xeon X5670 CPUs. This Phase I optimization was followed by further optimizations in Phase II, where single-node tests show an overall speedup of about 34 times on SummitDev and 7.9 times on Titan. The real physics tests on Summit machine showed impressive scaling properties that reaches roughly 50% efficiency on 928 nodes of Summit. The GPU + CPU speed up from purely CPU is over 20 times, leading to an unprecedented speed.

Original languageEnglish (US)
Title of host publicationAccelerator Programming Using Directives - 5th International Workshop, WACCPD 2018, Proceedings
EditorsGuido Juckeland, Sandra Wienke, Sunita Chandrasekaran
PublisherSpringer Verlag
Pages3-21
Number of pages19
ISBN (Print)9783030122737
DOIs
StatePublished - 2019
Event5th International Workshop on Accelerator Programming Using Directives, WACCPD 2018 - Dallas, United States
Duration: Nov 11 2018Nov 17 2018

Publication series

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

Conference

Conference5th International Workshop on Accelerator Programming Using Directives, WACCPD 2018
Country/TerritoryUnited States
CityDallas
Period11/11/1811/17/18

All Science Journal Classification (ASJC) codes

  • Theoretical Computer Science
  • General Computer Science

Keywords

  • Directives
  • Fusion plasma
  • GPU
  • Heterogeneous programming
  • Massively parallel computing
  • OpenACC
  • Particle in cell

Fingerprint

Dive into the research topics of 'Heterogeneous Programming and Optimization of Gyrokinetic Toroidal Code Using Directives'. Together they form a unique fingerprint.

Cite this