未验证 提交 26c44a86 编写于 作者: L Leo Chen 提交者: GitHub

revert flags_benchmark (#38005)

上级 7ccf67e5
...@@ -487,6 +487,14 @@ static void PreparedOpRunImpl( ...@@ -487,6 +487,14 @@ static void PreparedOpRunImpl(
op.Type(), outs, dev_ctx->GetPlace()); op.Type(), outs, dev_ctx->GetPlace());
} }
if (FLAGS_benchmark) {
dev_ctx->Wait();
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PADDLE_ENFORCE_GPU_SUCCESS(platform::GpuGetLastError());
VLOG(4) << "Operator(" << op.Type() << "): context wait and get last error";
#endif
}
/** /**
* [ Why need handle complex gradient to real gradient? ] * [ Why need handle complex gradient to real gradient? ]
* *
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册