Browse Source

fix memory leak in kernel_registry

tags/v1.1.0
hangangqiang 5 years ago
parent
commit
b45e7847c8
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/lite/src/kernel_registry.cc

+ 2
- 0
mindspore/lite/src/kernel_registry.cc View File

@@ -120,6 +120,8 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<Tensor *> &in_te
kernel->set_desc(key);
}
return kernel;
} else {
free(parameter);
}
return nullptr;
}


Loading…
Cancel
Save