POAS: A high-performance scheduling framework for exploiting Accelerator Level Parallelism

TL;DR

POAS is a flexible scheduling framework that transforms applications for efficient co-execution in Accelerator Level Parallelism environments, significantly improving performance in heterogeneous systems.

Contribution

It introduces a generic model that adapts applications for ALP co-execution without direct scheduling, demonstrated on matrix multiplication with notable speedups.

Findings

Achieved up to 45% speedup in matrix multiplication.

Validated POAS's flexibility and effectiveness in heterogeneous CPU/GPU/XPU systems.

Showed potential for future ALP-enabled computer architectures.

Abstract

Heterogeneous computing is becoming mainstream in all scopes. This new era in computer architecture brings a new paradigm called Accelerator Level Parallelism (ALP). In ALP, accelerators are used concurrently to provide unprecedented levels of performance and energy efficiency. To reach that, there are many problems to be solved, one of the most challenging being co-execution. This paper develops a scheduling framework called POAS, a general method for providing co-execution to generic applications. Unlike other scheduling approaches, POAS does not directly schedule applications. Instead, it is a generic model that transforms any application to make it suitable for co-execution, so that it can be executed in ALP environments. Our proposal is composed of four differentiated steps: predict, optimize, adapt and schedule. During these phases, different modifications are implemented in the application to make it suitable to be executed in ALP environments. In this work we also apply our framework to a matrix multiplication case study, outlining the critical and most important steps to port the application with POAS. We evaluate our POAS-based implementation for matrix multiplication on a CPU/GPU/XPU environment using CPU cores, CUDA cores and tensor cores (XPU). Our experiments prove that co-execution in the studied scenario can benefit from ALP, yielding speedups of up to 45% with respect to using only one accelerator. The proven flexibility and potential of POAS make it an excellent candidate to reach ALP in future computer systems.