Increase the default kernel max cache size to 1536 (#370)

microsoft · May 2, 2022 · 628dbd9 · 628dbd9
1 parent 6018b72
commit 628dbd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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();