diff --git a/vta/Jenkinsfile b/vta/Jenkinsfile index 90dfc27af434aa8299f7b8faae1871d0c62d7c03..a1f59206f29c04654f42c5d9e2af7e898cc2beca 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 5408f3b950b19dc681c86cb79627a5bf79c82a32..cd3a490ccb6f3ca1460f0f04a5fc0650a62b13fe 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 0000000000000000000000000000000000000000..4555a739a8d812cbefe83bbf0440ee2b8f0c3d0c --- /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 c56e9a444ac0fa5c3b67da3eaa3721ada9180071..37c2e7f3479206db9402fa3aed44bdad438c8bb4 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 8f8b4804bdccc6aad79ab683e2db3be27ef8ab05..7fac9f692c8add4f48864b9c4abada5081cedc2d 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