From b815c4e6fe65e66f5b67beebb17bd566c1406982 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Fri, 26 Jun 2020 06:40:20 +0000 Subject: [PATCH] fix str --- python/tvm/relay/backend/graph_runtime_codegen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/backend/graph_runtime_codegen.py b/python/tvm/relay/backend/graph_runtime_codegen.py index 8210f27732be3..1d718de1f6881 100644 --- a/python/tvm/relay/backend/graph_runtime_codegen.py +++ b/python/tvm/relay/backend/graph_runtime_codegen.py @@ -85,7 +85,7 @@ def codegen(self, func): param_names = self._list_params_name() params = {} for key in param_names: - arr = self._get_param_by_name(key) + arr = self._get_param_by_name(str(key)) param = empty(arr.shape, dtype=arr.dtype, ctx=arr.ctx) arr.copyto(param) params[key] = param