From 296838a10b27116c88faa9d2699d4fd194c238e6 Mon Sep 17 00:00:00 2001
From: "Durandin, Pavel" <pavel.durandin@intel.com>
Date: Fri, 18 Oct 2024 10:50:17 +0400
Subject: [PATCH] [GPU] Fix oneDnn cache leak

---
 src/plugins/intel_gpu/src/plugin/plugin.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/plugins/intel_gpu/src/plugin/plugin.cpp b/src/plugins/intel_gpu/src/plugin/plugin.cpp
index a28dfd72b1e9bc..671a81c3b73f01 100644
--- a/src/plugins/intel_gpu/src/plugin/plugin.cpp
+++ b/src/plugins/intel_gpu/src/plugin/plugin.cpp
@@ -164,7 +164,7 @@ Plugin::Plugin() {
     // update oneDnn cache
 #ifdef ENABLE_ONEDNN_FOR_GPU
 #if (ENABLE_ONEDNN_FOR_GPU)
-    dnnl::set_primitive_cache_capacity(1024);
+    //dnnl::set_primitive_cache_capacity(1024);
 #endif
 #endif
 }
@@ -174,6 +174,7 @@ Plugin::~Plugin() {
 #ifdef ENABLE_ONEDNN_FOR_GPU
 #if (ENABLE_ONEDNN_FOR_GPU)
     dnnl::set_primitive_cache_capacity(0);
+    dnnl::set_primitive_cache_capacity(1024);
 #endif
 #endif
 }