ir_params_sync_release (#49051)

502f5e88 · xiaoxiaohehe001 · GitHub · a1d6e2a8 · 502f5e88
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/inference/analysis/passes/ir_params_sync_among_devices_pass.cc ...ence/analysis/passes/ir_params_sync_among_devices_pass.cc +1 -0

未找到文件。
--- 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 {