From 152eef091aaae0ad1854aec07652208c836fef61 Mon Sep 17 00:00:00 2001 From: Serge Panev Date: Thu, 6 Aug 2020 18:01:14 -0700 Subject: [PATCH] Fix a bug in MXNet-TensorRT Signed-off-by: Serge Panev --- src/operator/subgraph/tensorrt/tensorrt-inl.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/operator/subgraph/tensorrt/tensorrt-inl.h b/src/operator/subgraph/tensorrt/tensorrt-inl.h index b35a1715000e..3119d31fa909 100644 --- a/src/operator/subgraph/tensorrt/tensorrt-inl.h +++ b/src/operator/subgraph/tensorrt/tensorrt-inl.h @@ -302,7 +302,7 @@ class TensorrtProperty : public SubgraphProperty { TRTParam param; std::ostringstream params_oss; for (auto ¶m_name : new_sym.ListInputNames(nnvm::Symbol::kAll)) { - NDArray *cache; + NDArray *cache = nullptr; auto it_args = in_args_dict.find(param_name); if (it_args != in_args_dict.end()) { cache = it_args->second;