From ded54e73afe6ba9f0baa381cba2b91b947c065e2 Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Wed, 23 Jun 2021 20:53:47 +0800 Subject: [PATCH] Fix Guard for variable release --- ge/graph/load/model_manager/model_manager.cc | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ge/graph/load/model_manager/model_manager.cc b/ge/graph/load/model_manager/model_manager.cc index cae828d6..99793252 100755 --- a/ge/graph/load/model_manager/model_manager.cc +++ b/ge/graph/load/model_manager/model_manager.cc @@ -1394,17 +1394,19 @@ Status ModelManager::LaunchKernelCustAicpuSo(const string &kernel_name) { return SUCCESS; } - vector allocated_mem; - rtError_t status; rtStream_t stream = nullptr; + vector allocated_mem; std::function callback = [&]() { for (auto mem : allocated_mem) { GE_CHK_RT(rtFree(mem)); } - GE_CHK_RT(rtStreamDestroy(stream)); + if (stream != nullptr) { + GE_CHK_RT(rtStreamDestroy(stream)); + } }; GE_MAKE_GUARD(release, callback); + rtError_t status; vector v_cust_so; void *args = nullptr;