matmap 0.1.0

MatMap: A Modular, Automatable, Tunable Mapper for Accelerator Programming

Matmap: A Modular, Automatable, Tunable Mapper for Accelerator Programming

A representation for higher-level transforms, currently targeting EXO code.

Setup

This has been extensively tested on Python 3.9.7. Python versions 3.7 and earlier are not supported as Exo requires several newer language features not available. If you are on a system with an outdated version of python, we recommend using pyenv to install a new version of pyenv.

git clone https://github.com/gdinh/matmap.git
python -m venv $HOME/.venv/matmap
source $HOME/.venv/matmap/bin/activate
python -m pip install --upgrade pip
python -m pip install -e ./matmap

Further documentation and demo notebooks can be found in the docs directory.

Project status:

Working:

• Tiling schedule (including automatic generation of tiles for projective nested loops)
• Reordering schedule

In progress:

• GPTune integration
• CoSA transform

Next TODOs:

• HBL autotiling for CNNs
• Code specialization for variable sized bounds