From 502f5e88e73b93b22f2ec6c733d608c803a64abe Mon Sep 17 00:00:00 2001
From: xiaoxiaohehe001 <49090790+xiaoxiaohehe001@users.noreply.github.com>
Date: Thu, 22 Dec 2022 16:15:36 +0800
Subject: [PATCH] ir_params_sync_release (#49051)

---
 .../analysis/passes/ir_params_sync_among_devices_pass.cc         | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/fluid/inference/analysis/passes/ir_params_sync_among_devices_pass.cc b/paddle/fluid/inference/analysis/passes/ir_params_sync_among_devices_pass.cc
index e3241d78e6..f325f2dec6 100644
--- a/paddle/fluid/inference/analysis/passes/ir_params_sync_among_devices_pass.cc
+++ b/paddle/fluid/inference/analysis/passes/ir_params_sync_among_devices_pass.cc
@@ -232,6 +232,7 @@ void IrParamsSyncAmongDevicesPass::RunImpl(Argument *argument) {
     CopyParamsToCustomDevice(argument);
   }
 #endif
+  paddle::memory::Release(platform::CPUPlace());
 }
 
 std::string IrParamsSyncAmongDevicesPass::repr() const {
-- 
GitLab