Skip to content

Commit

Permalink
Explain how to generate module library in Quick Start tutorial (apac…
Browse files Browse the repository at this point in the history
…he#323)

* Explain how to generate module library

* Small fix
  • Loading branch information
kevinthesun authored and sergei-mironov committed Aug 8, 2018
1 parent 47ff210 commit 5271997
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions nnvm/tutorials/define_and_compile_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,9 @@
# in an optimized runtime module for model serving.
#
# We'll first compile for Nvidia GPU.
# To generate the module library, TVM will first transfer graph IR into lower
# intrinsic IR for the specified target backend, which is CUDA in this example.
# Then target backend will generate module library.

opt_level = 0
target = tvm.target.cuda()
Expand Down

0 comments on commit 5271997

Please sign in to comment.