Skip to main navigation Skip to search Skip to main content

Productivity, portability, performance: data-centric python

  • Alexandros Nikolaos Ziogas
  • , Timo Schneider
  • , Tal Ben Nun
  • , Alexandru Calotoiu
  • , Tiziano De Matteis
  • , Johannes De Fine Licht
  • , Luca Lavarini
  • , Torsten Hoefler

Research output: Chapter in Book / Report / Conference proceedingConference contributionAcademicpeer-review

Abstract

Python has become the de facto language for scientific computing. Programming in Python is highly productive, mainly due to its rich science-oriented software ecosystem built around the NumPy module. As a result, the demand for Python support in High Performance Computing (HPC) has skyrocketed. However, the Python language itself does not necessarily offer high performance. In this work, we present a workflow that retains Pythons high productivity while achieving portable performance across different architectures the workflows key features are HPC-oriented language extensions and a set of automatic optimizations powered by a data-centric intermediate representation. We show performance results and scaling across CPU, GPU, FPGA, and the Piz Daint supercomputer (up to 23,328 cores), with 2.47x and 3.75x speedups over previous-best solutions, first-ever Xilinx and Intel FPGA results of annotated Python, and up to 93.16% scaling efficiency on 512 nodes.
Original languageEnglish
Title of host publicationProceedings of SC 2021
Subtitle of host publicationThe International Conference for High Performance Computing, Networking, Storage and Analysis
PublisherIEEE Computer Society
Pages1-13
Number of pages13
ISBN (Electronic)9781450384421
DOIs
Publication statusPublished - 2021
Externally publishedYes
Event33rd International Conference for High Performance Computing, Networking, Storage and Analysis: Science and Beyond, SC 2021 - Virtual, Online, United States
Duration: 14 Nov 202119 Nov 2021

Publication series

NameInternational Conference for High Performance Computing, Networking, Storage and Analysis, SC
ISSN (Print)2167-4329
ISSN (Electronic)2167-4337

Conference

Conference33rd International Conference for High Performance Computing, Networking, Storage and Analysis: Science and Beyond, SC 2021
Country/TerritoryUnited States
CityVirtual, Online
Period14/11/2119/11/21

Funding

This project received funding from the European Research Council (ERC) under the European Union’s Horizon 2020 program (grant agreements DAPP No. 678880, EPiGRAM-HS No. 801039, and DEEP-SEA No.955606). The Swiss National Science Foundation supports Tal Ben-Nun (Ambizione Project No. 185778). The authors would like to thank Mark Klein and the Swiss National Supercomputing Centre (CSCS), Paderborn University (DaceML-FPGA project), and Xilinx (XACC program) for support and access to computational resources.

FundersFunder number
DEEP-SEA
Swiss National Supercomputing Centre
Horizon 2020 Framework Programme955606
European Research Council
Schweizerischer Nationalfonds zur Förderung der Wissenschaftlichen Forschung185778
Horizon 2020801039, 678880
Universität Paderborn

    Fingerprint

    Dive into the research topics of 'Productivity, portability, performance: data-centric python'. Together they form a unique fingerprint.

    Cite this