From 4d061299649ee420afbb87235418959a65cb0968 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Wed, 30 Jan 2019 03:46:53 +0800 Subject: [PATCH] [BugFix] Copy intermediate result in debug runtime (#2520) --- 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):