提交 bff0fc61 编写于 作者: M Megvii Engine Team

fix(mge/interpreter): fix outputs check on async level0

GitOrigin-RevId: 46c6a2ce7062d2ee0920c2e1ea9047536f475ac5
上级 142f31a8
......@@ -132,8 +132,9 @@ SmallVector<void*> ChannelImpl::apply_op(
} else if (m_async_level == 0) {
sync();
// check device error
for (auto&& oup : cmd.outputs) {
oup->ptr->comp_node().sync();
for (auto&& oup : outputs) {
auto info = reinterpret_cast<TensorInfo*>(oup);
info->ptr->comp_node().sync();
}
}
return outputs;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册