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

fix(imperative): fix the tensor info lock in worker

GitOrigin-RevId: d2fb4f8cc4cce45df0c11221b2b5554f9c41311c
上级 2694ff81
......@@ -710,6 +710,7 @@ ChannelImpl::~ChannelImpl() {
void ChannelImpl::produce_tensor(TensorInfo* dest, TensorPtr ptr) {
auto& state = get_worker_state();
MGB_LOCK_GUARD(m_mutex);
MGB_LOCK_GUARD(m_info_spin);
m_dtr.update_used_time(dest);
MGB_RECORD_EVENT(
TensorProduceEvent, dest->id, ptr->layout(), ptr->comp_node(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册