Install and use NNVM Compiler

machine-learning
tvm-stack
nnvm
nnvm-compiler
compiler

(Team) #1

NNVM compiler is a graph compiler for the TVM Stack that takes in models in NNVM Intermediate Representation format and compiles them for various backends such as LLVM, METAL, CUDA and others.


NNVM Compiler supports the following targets:
  • LLVM
  • CUDA
  • METAL
  • OpenCL


NNVM Compiler has four optimization levels:
  • 0
  • 1
  • 2 (default optimization level)
  • 3


NNVM Compiler has several pass optimizations that you may apply such as:
  • AlterOpLayout
  • FoldScaleAxis
  • OpFusion
  • PrecomputePrune
  • SimplifyInference


Read the article for further details, installation guide and a code example
This is a companion discussion topic for the original entry at http://iq.opengenus.org/install-use-nnvm-compiler/