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