Browse Source

Fix Guard for variable release

tags/v1.3.0
zhangxiaokun 4 years ago
parent
commit
ded54e73af
1 changed files with 5 additions and 3 deletions
  1. +5
    -3
      ge/graph/load/model_manager/model_manager.cc

+ 5
- 3
ge/graph/load/model_manager/model_manager.cc View File

@@ -1394,17 +1394,19 @@ Status ModelManager::LaunchKernelCustAicpuSo(const string &kernel_name) {
return SUCCESS;
}

vector<void *> allocated_mem;
rtError_t status;
rtStream_t stream = nullptr;
vector<void *> allocated_mem;
std::function<void()> 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<CustAicpuSoBuf> v_cust_so;
void *args = nullptr;



Loading…
Cancel
Save