/pyvlova

Yet another Polyhedra Compiler for DeepLearning

Primary LanguagePythonApache License 2.0Apache-2.0

Pyvlova

A Simple Polyhedral Compiler for NN

Benchmark

Requirements

  • tvm
  • sympy, numpy

Build extensions (Build ISL)

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

python setup.py build_ext

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