From c12691a20370b0d30b0c84926694a26fc266ea11 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Thu, 24 May 2018 10:36:46 -0700
Subject: [PATCH] [INIT] Allow proper throw in compiler (#39)
---
vta/python/vta/__init__.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/vta/python/vta/__init__.py b/vta/python/vta/__init__.py
index c2dc7c30d..926d73649 100644
--- a/vta/python/vta/__init__.py
+++ b/vta/python/vta/__init__.py
@@ -6,15 +6,17 @@ configure the hardware Environment and access remote through RPC
"""
from __future__ import absolute_import as _abs
-__version__ = "0.1.0"
+import sys
from .bitstream import get_bitstream_path, download_bitstream
from .environment import get_env, Environment
from .rpc_client import reconfig_runtime, program_fpga
-try:
+__version__ = "0.1.0"
+
+# do not import nnvm/topi when running vta.exec.rpc_server
+# to maintain minimum dependency on the board
+if sys.argv[0] not in ("-c", "-m"):
from . import top
from .build_module import build_config, lower, build
from . import graph
-except (ImportError, RuntimeError):
- pass
--
GitLab