Pyvlova

A Simple Polyhedral Compiler for NN

Benchmark

• ResNet-18
• MobileNetV2

Requirements

• tvm
• sympy, numpy

Build extensions (Build ISL)

git submodule update --init --recursive --progress 3rdparty/isl

python setup.py build_ext

ISL

• isl (ChieloNewctle/isl)

Requirements

• llvm and clang
• libgmp-dev
• libclang-*-dev, such as libclang-10-dev, IMPORTANT for building the python interface
• automake, autoconf, libtool
• pkg-config

Run tests

To test CUDA kernels generated by pyvlova:

python -m unittest test.test_op