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;