[GPU] Fix oneDnn cache leak

p-durandin · Oct 18, 2024 · 296838a · 296838a
1 parent 9139472
commit 296838a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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
 }