Benchmarking the performance of GPT-2 type applications on GPU-accelerated computing resources

In this report we look at eight GPU-accelerated systems representing across-section of the available Tier 2 HPC systems in the UK. These includea mixture of GPU-accelerated platforms from NVIDIA, AMD and Intel. For each system, we perform a set of benchmarking experiments by training a GPT-2 model using a mixture of parameters and hyperparameters. These include model size, number of GPUs, floating point data type, training datasize, distribution strategies and batch size. Our interest is in performance measured by time taken to complete an epoch of training rather than convergence speed. We also measure memory usage. The overall aim is to compare systems in order to provide researchers intending to perform AI training with some benchmarks for what to expect in terms of training speed for a modelin a non-optimised real-world scenario.

Institutions

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2024-08-21
DOI
https://doi.org/10.5281/zenodo.13349540
Primary Topic
Parallel Computing and Optimization Techniques
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Benchmarking the performance of GPT-2 type applications on GPU-accelerated computing resources

Zenodo (CERN European Organization for Nuclear Research)
Parallel Computing and Optimization Techniques
article

Benchmarking the performance of GPT-2 type applications on GPU-accelerated computing resources

article en

Abstract

In this report we look at eight GPU-accelerated systems representing across-section of the available Tier 2 HPC systems in the UK. These includea mixture of GPU-accelerated platforms from NVIDIA, AMD and Intel. For each system, we perform a set of benchmarking experiments by training a GPT-2 model using a mixture of parameters and hyperparameters. These include model size, number of GPUs, floating point data type, training datasize, distribution strategies and batch size. Our interest is in performance measured by time taken to complete an epoch of training rather than convergence speed. We also measure memory usage. The overall aim is to compare systems in order to provide researchers intending to perform AI training with some benchmarks for what to expect in terms of training speed for a modelin a non-optimised real-world scenario.

Zenodo (CERN European Organization for Nuclear Research)
The Alan Turing Institute (GB)
Openalex Percentile: Top 59%
Parallel Computing and Optimization Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.