reduce default value of cudnn workspace size, test=develop (#19780)

32b1151f · Zeng Jinle · GitHub · 52673956 · 32b1151f
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/platform/cudnn_workspace_helper.h paddle/fluid/platform/cudnn_workspace_helper.h +1 -1

未找到文件。
--- a/paddle/fluid/platform/cudnn_workspace_helper.h
+++ b/paddle/fluid/platform/cudnn_workspace_helper.h
@@ -17,7 +17,7 @@
 namespace paddle {
 namespace platform {

-static constexpr int kDefaultConvWorkspaceSizeLimitMB = 4096;
+static constexpr int kDefaultConvWorkspaceSizeLimitMB = 512;

 }  // namespace platform
 }  // namespace paddle