SC22 Proceedings

The International Conference for High Performance Computing, Networking, Storage, and Analysis

Technical Papers Archive

Parla: A Python Orchestration System for Heterogeneous Architectures

Authors: Hochan Lee and William Ruys (University of Texas); Ian Henriksen (University of Texas, Jabberwock Technologies Inc); Arthur Peters (University of Texas, Katana Graph Inc); Yineng Yan, Sean Stephens, Bozhi You, and Henrique Fingler (University of Texas); Martin Burtscher (Texas State University); Milos Gligoric (University of Texas); Karl Schulz and Keshav Pingali (University of Texas, Oden Institute); Christopher J. Rossbach and Mattan Erez (University of Texas); and George Biros (University of Texas, Oden Institute)

Abstract: Python’s ease of use and rich collection of numeric libraries make it an excellent choice for rapidly developing scientific applications. However, composing these libraries to take advantage of complex heterogeneous nodes is still difficult. To simplify writing multi-device code, we created Parla, a heterogeneous task-based programming framework that fully supports Python’s scientific programming stack. Parla’s API is based on Python decorators and allows users to wrap code in Parla tasks for parallel execution. Parla arrays enable automatic movement of data between devices. The Parla runtime handles resource-aware mapping, scheduling, and execution of tasks. Compared to other Python tasking systems, Parla is unique in its parallelization of tasks within a single process, its GPU context and resource-aware runtime, and its design around gradual adoption to provide easy migration of and integration into existing Python applications. We show that Parla can achieve performance competitive with hand-optimized code while improving ease of development.

Back to Technical Papers Archive Listing