arXiv · 2404.10536
Benchmarking Machine Learning Applications on Heterogeneous Architecture using Reframe
Abstract
With the rapid increase in machine learning workloads performed on HPC systems, it is beneficial to regularly perform machine learning specific benchmarks to monitor performance and identify issues. Furthermore, as part of the Edinburgh International Data Facility, EPCC currently hosts a wide range of machine learning accelerators including Nvidia GPUs, the Graphcore Bow Pod64 and Cerebras CS-2, which are managed via Kubernetes and Slurm. We extended the Reframe framework to support the Kubernetes scheduler backend, and utilise Reframe to perform machine learning benchmarks, and we discuss the preliminary results collected and challenges involved in integrating Reframe across multiple platforms and architectures.
Explore related subjects
Keep this discovery
Christopher Rae, Joseph K. L. Lee, James Richings, Michele Weiland. 2024-04-16. Benchmarking Machine Learning Applications on Heterogeneous Architecture using Reframe. https://arxiv.org/abs/2404.10536
Cite the original work for its findings. Save a collection to share your selection of sources.