Skip to content

Commit be7ada2

Browse files
YeelandXYeelandX
andauthored
adam op optimization (PaddlePaddle#51)
* update XDNN version 20240125: get fc hit rate * adam op optimization --------- Co-authored-by: YeelandX <[email protected]>
1 parent 4879537 commit be7ada2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/operators/optimizers/adam_op_xpu.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ class AdamOpXPUKernel : public framework::OpKernel<T> {
176176
epsilon,
177177
param.numel());
178178

179-
xpu_wait(dev_ctx.x_context()->xpu_stream);
179+
// xpu_wait(dev_ctx.x_context()->xpu_stream);
180180
PADDLE_ENFORCE_EQ(
181181
r == xpu::Error_t::SUCCESS,
182182
true,
@@ -225,7 +225,7 @@ class AdamOpXPUKernel : public framework::OpKernel<T> {
225225
r,
226226
XPUAPIErrorMsg[r]));
227227

228-
xpu_wait(dev_ctx.x_context()->xpu_stream);
228+
// xpu_wait(dev_ctx.x_context()->xpu_stream);
229229
}
230230
}
231231
} else if (grad_var->IsType<phi::SelectedRows>()) {

0 commit comments

Comments
 (0)