Tiramisu is a compiler for expressing fast and portable data parallel computations. It provides a simple C++ API for expressing algorithms (Tiramisu expressions
) and how these algorithms should be optimized by the compiler. Tiramisu can be used in areas such as linear and tensor algebra, deep learning, image processing, stencil computations and machine learning.
The Tiramisu compiler is based on the polyhedral model thus it can express a large set of loop optimizations and data layout transformations. Currently it targets (1) multicore X86 CPUs, (2) Nvidia GPUs, (3) Xilinx FPGAs (Vivado HLS) and (4) distributed machines (using MPI). It is designed to enable easy integration of code generators for new architectures.
The following is an example of a Tiramisu program specified using the C++ API.
// C++ code with a Tiramisu expression.
#include "tiramisu/tiramisu.h"
using namespace tiramisu;
void generate_code()
{
// Specify the name of the function that you want to create.
tiramisu::init("foo");
// Declare two iterator variables (i and j) such that 0<=i<100 and 0<=j<100.
var i("i", 0, 100), j("j", 0, 100);
// Declare a Tiramisu expression (algorithm) that is equivalent to the following C code
// for (i=0; i<100; i++)
// for (j=0; j<100; j++)
// C(i,j) = 0;
computation C({i,j}, 0);
// Specify optimizations
C.parallelize(i);
C.vectorize(j, 4);
buffer b_C("b_C", {100, 100}, p_int32, a_output);
C.store_in(&b_C);
// Generate code
C.codegen({&b_C}, "generated_code.o");
}
This section provides a description of how to build Tiramisu. The installation instructions below have been tested on Linux Ubuntu (18.04) and MacOS (13.0.1) but should work on other Linux and MacOS versions.
-
OpenMPI and OpenSSh: if you want to generate and run distributed code (MPI).
-
CUDA Toolkit: if you want to generate and run CUDA code.
-
Python 3.8 or higher if you want to use the python bindings. (Along with Pybind 2.10.2, Cython, and Numpy).
There are 3 ways to build Tiramisu:
- From spack, which will build everything from source for you.
- From source, but using system package managers for dependencies.
- Purely from source with our install script.
The last two only differ only in how they setup the dependenies.
Install spack and then run:
spack install tiramisu
There are two steps:
- Install the dependencies (either using Homebrew or using Apt).
- Use Cmake to build Tiramisu.
If you are on MacOS and using Homebrew, you can run the following commands to setup the dependencies:
brew install cmake
brew install llvm@14
brew install halide
brew install isl
brew link halide
brew link isl
If any of these ask you to update your path, do so. For example, using the following command, you can find the isl include and library directories:
brew info isl
ISL_INCLUDE_DIRECTORY=..
ISL_LIB_DIRECTORY=..
If you are on Ubuntu/Debian, you can use apt to setup the dependencies:
wget https://apt.llvm.org/llvm.sh
chmod +x llvm.sh
sudo ./llvm.sh 14 all
sudo apt-get install liblld-14-dev llvm-14-runtime
sudo apt-get install libllvm14 llvm-14-dev
sudo apt-get install llvm14-*
sudo apt-get install halide
sudo apt-get install libisl-dev
Using the following command, you can find the isl include and library directories:
dpkg -L libisl-dev
ISL_INCLUDE_DIRECTORY=..
ISL_LIB_DIRECTORY=..
- Get Tiramisu
git clone https://github.com/Tiramisu-Compiler/tiramisu.git
cd tiramisu
mkdir build
-
Setup the configure.cmake. In particular, choose if you want to use a GPU or MPI setup. Choose if you want to use the python bindings. Choose if you want to us the auto scheduler. You may need to add other options to support these.
-
Configure:
cmake . -B build -DISL_LIB_DIRECTORY=$ISL_LIB_DIRECTORY -DISL_INCLUDE_DIRECTORY=$ISL_INCLUDE_DIRECTORY -DPython3_EXECUTABLE=`which python3`
If you want to install, add CMAKE_INSTALL_PREFIX
. If you are installing the python bindings, add Tiramisu_INSTALL_PYTHONDIR
to tell Tiramisu where to place a python package. You will need add these install locations to the relevant path variables such as PYTHONPATH
and LD_LIBRARY_PATH
.
- Build:
cmake --build build
You can also install if you want via cmake --install
.
There are two steps:
- Install the dependencies using our script.
- Use Cmake to build Tiramisu.
- Get Tiramisu
git clone https://github.com/Tiramisu-Compiler/tiramisu.git
cd tiramisu
- Get and install Tiramisu submodules (ISL, LLVM and Halide). This step may take between few minutes to few hours (downloading and compiling LLVM is time consuming).
./utils/scripts/install_submodules.sh <TIRAMISU_ROOT_DIR>
- Note: Make sure `<TIRAMISU_ROOT_DIR>` is absolute path!
-
Optional: configure the tiramisu build by editing
configure.cmake
. Needed only if you want to generate MPI or GPU code, run the BLAS benchmarks, or if you want to build the autoscheduler module. A description of what each variable is and how it should be set is provided in comments inconfigure.cmake
.- To use the GPU backend, set
USE_GPU
toTRUE
. If the CUDA library is not found automatically while building Tiramisu, the user will be prompt to provide the path to the CUDA library. - To use the distributed backend, set
USE_MPI
toTRUE
. If the MPI library is not found automatically, set the following variables: MPI_INCLUDE_DIR, MPI_LIB_DIR, and MPI_LIB_FLAGS. - To build the autoscheduler module, set
USE_AUTO_SCHEDULER
toTRUE
.
- To use the GPU backend, set
-
Add Halide's cmake to the
CMAKE_PREFIX_PATH
:
export CMAKE_PREFIX_PATH=<TIRAMISU_ROOT_DIR>/3rdParty/Halide/build/:$CMAKE_PREFIX_PATH
- Build the main Tiramisu library
mkdir build
cd build
cmake ..
cmake --build .
- If you want to build the autoscheduler module, set
USE_AUTO_SCHEDULER
toTRUE
inconfigure.cmake
, and after building Tiramisu :
make tiramisu_auto_scheduler
Users can use the Tiramisu virtual machine disk image. The image is created using virtual box (5.2.12) and has Tiramisu already pre-compiled and ready for use. It was compiled using the same instructions in this README file.
Once you download the image, unzip it and use virtual box to open the file 'TiramisuVM.vbox'.
Once the virtual machine has started, open a terminal, then go to the Tiramisu directory
cd /home/b/tiramisu/
If asked for a username/password
Username:b
Password:b
- Build Tiramisu.
- Read the Tutorials.
- Read the Tiramisu Paper.
- Subscribe to Tiramisu mailing list.
- Read the compiler internal documentation (if you want to contribute to the compiler).
To run all the tests, assuming you are in the build/ directory
make test
or
ctest
To run only one test (test_01 for example)
ctest -R 01
This will compile and run the code generator and then the wrapper.
To view the output of a test pass the --verbose
option to ctest
.