From 012697a333ebb39f4724baf73e7e6fef8dd48d31 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 90dfc27af..a1f59206f 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 5408f3b95..cd3a490cc 100644 --- a/vta/README.md +++ b/vta/README.md @@ -1,6 +1,6 @@ VTA: Open, Modular, Deep Learning Accelerator Stack =================================================== - +[](http://mode-gpu.cs.washington.edu:8080/job/uwsaml/job/vta/job/master/) [](./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 000000000..4555a739a --- /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 c56e9a444..37c2e7f34 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 8f8b4804b..7fac9f692 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 -- GitLab