From 32b1151f5ee37c51663737a383ad44f54ff2698c Mon Sep 17 00:00:00 2001 From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com> Date: Mon, 16 Sep 2019 11:08:38 +0800 Subject: [PATCH] reduce default value of cudnn workspace size, test=develop (#19780) --- paddle/fluid/platform/cudnn_workspace_helper.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/platform/cudnn_workspace_helper.h b/paddle/fluid/platform/cudnn_workspace_helper.h index 58f76e3128..29abdc72e2 100644 --- 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