From 628dbd9f146810df93525fa4e1553a1de7962c9e Mon Sep 17 00:00:00 2001 From: Patrice Vignola Date: Mon, 2 May 2022 11:22:50 -0700 Subject: [PATCH] Increase the default kernel max cache size to 1536 (#370) --- tensorflow/core/common_runtime/dml/dml_kernel_manager.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/core/common_runtime/dml/dml_kernel_manager.h b/tensorflow/core/common_runtime/dml/dml_kernel_manager.h index 0b2a6406b7d..c6eb49efd30 100644 --- a/tensorflow/core/common_runtime/dml/dml_kernel_manager.h +++ b/tensorflow/core/common_runtime/dml/dml_kernel_manager.h @@ -49,7 +49,7 @@ class NoOpInitializationHelper; class DmlKernelManager { public: // Can be overridden by the TF_DIRECTML_KERNEL_CACHE_SIZE environment variable - static constexpr size_t kDefaultMaxCacheSize = 1024; + static constexpr size_t kDefaultMaxCacheSize = 1536; DmlKernelManager();