Skip to content

Commit

Permalink
fix jenkins (apache#31)
Browse files Browse the repository at this point in the history
  • Loading branch information
tqchen committed Jul 12, 2018
1 parent 395f4c3 commit 012697a
Show file tree
Hide file tree
Showing 5 changed files with 42 additions and 4 deletions.
4 changes: 2 additions & 2 deletions vta/Jenkinsfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion vta/README.md
Original file line number Diff line number Diff line change
@@ -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.
Expand Down
28 changes: 28 additions & 0 deletions vta/docs/api/python/index.rst
Original file line number Diff line number Diff line change
@@ -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
7 changes: 6 additions & 1 deletion vta/python/vta/__init__.py
Original file line number Diff line number Diff line change
@@ -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"
Expand Down
5 changes: 5 additions & 0 deletions vta/tests/scripts/task_python_docs.sh
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
#!/bin/bash
cd nnvm/tvm
make cython
make cython3
cd ../../

mkdir -p docs/_build/html
# C++ doc
make doc
Expand Down

0 comments on commit 012697a

Please sign in to comment.