Browse Source

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

GitOrigin-RevId: 46c6a2ce70
release-1.2
Megvii Engine Team 4 years ago
parent
commit
bff0fc6172
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      imperative/src/impl/interpreter_impl.cpp

+ 3
- 2
imperative/src/impl/interpreter_impl.cpp View File

@@ -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;


Loading…
Cancel
Save