未验证 提交 1a8f20c6 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #6284 from qingqing01/cuda_sync

Remove the CUDA stream synchronization between each operator.
...@@ -426,9 +426,6 @@ void OperatorWithKernel::Run(const Scope& scope, ...@@ -426,9 +426,6 @@ void OperatorWithKernel::Run(const Scope& scope,
} }
kernel_iter->second->Compute(ctx); kernel_iter->second->Compute(ctx);
// throws errors if have.
dev_ctx.Finish();
} }
OpKernelType OperatorWithKernel::GetKernelType( OpKernelType OperatorWithKernel::GetKernelType(
const ExecutionContext& ctx) const { const ExecutionContext& ctx) const {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册