Skip to content

ModelTC/pyvlova

Repository files navigation

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

About

Yet another Polyhedra Compiler for DeepLearning

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages