Restore cache capacity after cleanup

openvinotoolkit · Feb 9, 2024 · 8275c73 · 8275c73
1 parent b49f7ae
commit 8275c73
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/plugins/intel_gpu/src/plugin/plugin.cpp b/src/plugins/intel_gpu/src/plugin/plugin.cpp
@@ -166,6 +166,8 @@ Plugin::~Plugin() {
     // trigger earlier cache cleanup by setting its capacity to 0.
     // Related ticket: 106154.
     dnnl::set_primitive_cache_capacity(0);
+    const int default_cache_capacity = 1024;
+    dnnl::set_primitive_cache_capacity(default_cache_capacity);
 #endif
 }