From 70b4fd6045b7c8eb0f1c9d7339ef86b7637311d5 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Tue, 29 Jan 2019 14:32:10 +0800 Subject: [PATCH] [BugFix] Copy intermediate result in debug runtime --- python/tvm/contrib/debugger/debug_runtime.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/tvm/contrib/debugger/debug_runtime.py b/python/tvm/contrib/debugger/debug_runtime.py index d38ee6cf7982..505b5e259fcc 100644 --- a/python/tvm/contrib/debugger/debug_runtime.py +++ b/python/tvm/contrib/debugger/debug_runtime.py @@ -7,6 +7,7 @@ from tvm._ffi.base import string_types from tvm._ffi.function import get_global_func from tvm.contrib import graph_runtime +from tvm.ndarray import array from tvm.rpc import base as rpc_base from . import debug_result @@ -172,6 +173,7 @@ def _run_debug(self): num_outputs = self.debug_datum.get_graph_node_output_num(node) for j in range(num_outputs): out_tensor = self._get_output_by_layer(i, j) + out_tensor = array(out_tensor) self.debug_datum._output_tensor_list.append(out_tensor) def debug_get_output(self, node, out):