From c38146c9f8d5f59f97caac8d31639dcb5eb8555b Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Sat, 14 Apr 2018 21:53:53 -0700
Subject: [PATCH] fix jenkins (#31)

---
 vta/Jenkinsfile                       |  4 ++--
 vta/README.md                         |  2 +-
 vta/docs/api/python/index.rst         | 28 +++++++++++++++++++++++++++
 vta/python/vta/__init__.py            |  7 ++++++-
 vta/tests/scripts/task_python_docs.sh |  5 +++++
 5 files changed, 42 insertions(+), 4 deletions(-)
 create mode 100644 vta/docs/api/python/index.rst

diff --git a/vta/Jenkinsfile b/vta/Jenkinsfile
index 90dfc27af434a..a1f59206f29c0 100644
--- a/vta/Jenkinsfile
+++ b/vta/Jenkinsfile
@@ -4,8 +4,8 @@
 // See documents at https://jenkins.io/doc/book/pipeline/jenkinsfile/
 
 // nnvm libraries
-vta_lib += "lib/libvta.so, lib/libvta.so.json"
-vta_lib += ", nnvm/tvm/lib/libtvm.so, nnvm/tvm/lib/libtopi.so, nnvm/lib/libnnvm_compiler.so"
+vta_lib = "lib/libvta.so, lib/libvta.so.json"
+vta_lib += ", nnvm/tvm/lib/libtvm.so, nnvm/tvm/lib/libtvm_topi.so, nnvm/lib/libnnvm_compiler.so"
 
 
 // command to start a docker container
diff --git a/vta/README.md b/vta/README.md
index 5408f3b950b19..cd3a490ccb6f3 100644
--- a/vta/README.md
+++ b/vta/README.md
@@ -1,6 +1,6 @@
 VTA: Open, Modular, Deep Learning Accelerator Stack
 ===================================================
-
+[![Build Status](http://mode-gpu.cs.washington.edu:8080/buildStatus/icon?job=uwsaml/vta/master)](http://mode-gpu.cs.washington.edu:8080/job/uwsaml/job/vta/job/master/)
 [![GitHub license](http://dmlc.github.io/img/apache2.svg)](./LICENSE)
 
 VTA(versatile tensor accelerator) is an open-source deep learning accelerator stack.
diff --git a/vta/docs/api/python/index.rst b/vta/docs/api/python/index.rst
new file mode 100644
index 0000000000000..4555a739a8d81
--- /dev/null
+++ b/vta/docs/api/python/index.rst
@@ -0,0 +1,28 @@
+Python API
+==========
+
+This document contains the python API to VTA compiler toolchain.
+
+.. automodule:: vta
+
+Hardware Information
+--------------------
+
+.. autofunction:: vta.Environment
+.. autofunction:: vta.get_env
+
+RPC Utilities
+-------------
+
+.. autofunction:: vta.reconfig_runtime
+.. autofunction:: vta.program_fpga
+
+
+Compiler API
+------------
+We program VTA using TVM, so the compiler API in vta package
+is only a thin wrapper to provide VTA specific extensions.
+
+.. autofunction:: vta.build_config
+.. autofunction:: vta.build
+.. autofunction:: vta.lower
diff --git a/vta/python/vta/__init__.py b/vta/python/vta/__init__.py
index c56e9a444ac0f..37c2e7f347920 100644
--- a/vta/python/vta/__init__.py
+++ b/vta/python/vta/__init__.py
@@ -1,4 +1,9 @@
-"""TVM-based VTA Compiler Toolchain"""
+"""VTA Package is a TVM backend extension to support VTA hardwares
+
+Besides the compiler toolchain.
+It also include utility functions to
+configure the hardware Environment and  access remote through RPC
+"""
 from __future__ import absolute_import as _abs
 
 __version__ = "0.1.0"
diff --git a/vta/tests/scripts/task_python_docs.sh b/vta/tests/scripts/task_python_docs.sh
index 8f8b4804bdccc..7fac9f692c8ad 100755
--- a/vta/tests/scripts/task_python_docs.sh
+++ b/vta/tests/scripts/task_python_docs.sh
@@ -1,4 +1,9 @@
 #!/bin/bash
+cd nnvm/tvm
+make cython
+make cython3
+cd ../../
+
 mkdir -p docs/_build/html
 # C++ doc
 make doc