inducer / loopy

A code generator for array-based code on CPUs and GPUs
http://mathema.tician.de/software/loopy
MIT License
565 stars 71 forks source link
array code-generation code-generator code-optimization code-transformation cuda ispc loop-optimization multidimensional-arrays opencl performance performance-analysis prefix-sum python reduction scan scientific-computing

Loopy: Transformation-Based Generation of High-Performance CPU/GPU Code

.. image:: https://gitlab.tiker.net/inducer/loopy/badges/main/pipeline.svg :alt: Gitlab Build Status :target: https://gitlab.tiker.net/inducer/loopy/commits/main .. image:: https://github.com/inducer/loopy/workflows/CI/badge.svg?branch=main&event=push :alt: Github Build Status :target: https://github.com/inducer/loopy/actions?query=branch%3Amain+workflow%3ACI+event%3Apush .. image:: https://badge.fury.io/py/loopy.png :alt: Python Package Index Release Page :target: https://pypi.org/project/loopy/ .. image:: https://zenodo.org/badge/20281732.svg :alt: Zenodo DOI for latest release :target: https://zenodo.org/doi/10.5281/zenodo.10672274

Loopy lets you easily generate the tedious, complicated code that is necessary to get good performance out of GPUs and multi-core CPUs. Loopy's core idea is that a computation should be described simply and then transformed into a version that gets high performance. This transformation takes place under user control, from within Python.

It can capture the following types of optimizations:

Loopy targets array-type computations, such as the following:

It is not (and does not want to be) a general-purpose programming language.

Loopy is licensed under the liberal MIT license <https://en.wikipedia.org/wiki/MIT_License>_ and free for commercial, academic, and private use. All of Loopy's dependencies can be automatically installed from the package index after using::

pip install loopy

In addition, Loopy is compatible with and enhances pyopencl <https://mathema.tician.de/software/pyopencl>_.


Places on the web related to Loopy: