From d1f56fdcc6f981b3d51bd04fb2b1ed8857480c8f Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Fri, 4 Oct 2019 20:57:27 +0000 Subject: [PATCH 1/2] fix --- python/tvm/relay/backend/vm.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/backend/vm.py b/python/tvm/relay/backend/vm.py index 745ef63d3afe..ee0ee87501da 100644 --- a/python/tvm/relay/backend/vm.py +++ b/python/tvm/relay/backend/vm.py @@ -178,9 +178,13 @@ def compile(self, mod, target=None, target_host=None): """ target = _update_target(target) target_host = None if target_host == "" else target_host + if not target_host: + for device_type, tgt in target.items(): + if device_type.value == tvm.nd.cpu(0).device_type: + target_host = tgt + break if not target_host: target_host = "llvm" if tvm.module.enabled("llvm") else "stackvm" - target_host = tvm.target.create(target_host) self._compile(mod, target, target_host) return VirtualMachine(self._get_vm()) From 227f547a74df16f7234a07fab1a69f4cb52fbbb3 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Fri, 4 Oct 2019 21:01:21 +0000 Subject: [PATCH 2/2] tweak --- python/tvm/relay/backend/vm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/backend/vm.py b/python/tvm/relay/backend/vm.py index ee0ee87501da..251d3cc3463e 100644 --- a/python/tvm/relay/backend/vm.py +++ b/python/tvm/relay/backend/vm.py @@ -185,6 +185,7 @@ def compile(self, mod, target=None, target_host=None): break if not target_host: target_host = "llvm" if tvm.module.enabled("llvm") else "stackvm" + target_host = tvm.target.create(target_host) self._compile(mod, target, target_host) return VirtualMachine(self._get_vm())