forked from mindspore-Ecosystem/mindspore
!8009 fix memory leak in kernel_registry
Merge pull request !8009 from hangq/primitive
This commit is contained in:
commit
7f58cd214d
|
@ -120,6 +120,8 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<Tensor *> &in_te
|
||||||
kernel->set_desc(key);
|
kernel->set_desc(key);
|
||||||
}
|
}
|
||||||
return kernel;
|
return kernel;
|
||||||
|
} else {
|
||||||
|
free(parameter);
|
||||||
}
|
}
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue