Skip to content

T
tic
Commit 54051271 by Yao Wang Committed by Tianqi Chen
Options

Explain how to generate module library in Quick Start tutorial (#323) 

* Explain how to generate module library

* Small fix
parent f34e1744
Showing with 3 additions and 0 deletions
nnvm/tutorials/define_and_compile_model.py
......@@ -68,6 +68,9 @@ print(net.debug_str())
# 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()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment