Skip to content

Commit

Permalink
[GPU] Fix oneDnn cache leak
Browse files Browse the repository at this point in the history
  • Loading branch information
p-durandin committed Oct 18, 2024
1 parent 5537178 commit 4865ceb
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/plugins/intel_gpu/src/plugin/plugin.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -166,11 +166,9 @@ Plugin::Plugin() {
Plugin::~Plugin() {
// reset oneDNN cache to clean up cached primitives
#ifdef ENABLE_ONEDNN_FOR_GPU
#if (ENABLE_ONEDNN_FOR_GPU)
dnnl::set_primitive_cache_capacity(0);
dnnl::set_primitive_cache_capacity(1024);
#endif
#endif
}

std::shared_ptr<ov::ICompiledModel> Plugin::compile_model(const std::shared_ptr<const ov::Model>& model, const ov::AnyMap& orig_config) const {
Expand Down

0 comments on commit 4865ceb

Please sign in to comment.