Browse Source

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

GitOrigin-RevId: d2fb4f8cc4
master
Megvii Engine Team 2 years ago
parent
commit
1137ebf3b9
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/src/impl/interpreter/interpreter_impl.cpp

+ 1
- 0
imperative/src/impl/interpreter/interpreter_impl.cpp View File

@@ -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(),


Loading…
Cancel
Save