diff --git a/Jenkinsfile b/Jenkinsfile index 90dfc27af434..a1f59206f29c 100644 --- a/Jenkinsfile +++ b/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/README.md b/README.md index 5408f3b950b1..cd3a490ccb6f 100644 --- a/README.md +++ b/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/docs/api/python/index.rst b/docs/api/python/index.rst new file mode 100644 index 000000000000..4555a739a8d8 --- /dev/null +++ b/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/python/vta/__init__.py b/python/vta/__init__.py index c56e9a444ac0..37c2e7f34792 100644 --- a/python/vta/__init__.py +++ b/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/tests/scripts/task_python_docs.sh b/tests/scripts/task_python_docs.sh index 8f8b4804bdcc..7fac9f692c8a 100755 --- a/tests/scripts/task_python_docs.sh +++ b/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