Browse Source

fix(build/windows/macos): fix build caused by 40ca9ea80e

GitOrigin-RevId: b4868a08f6
release-1.1
Megvii Engine Team 4 years ago
parent
commit
dacc485407
2 changed files with 4 additions and 3 deletions
  1. +2
    -1
      imperative/src/impl/profiler.cpp
  2. +2
    -2
      imperative/src/include/megbrain/imperative/profiler.h

+ 2
- 1
imperative/src/impl/profiler.cpp View File

@@ -83,7 +83,8 @@ size_t TensorRecorder::record_tensor(const TensorPtr& tensor) {
return id; return id;
} else { } else {
auto id = m_next_id++; auto id = m_next_id++;
m_tensor_map.insert({tensor.get(), {std::weak_ptr{tensor}, id}});
m_tensor_map.insert(
{tensor.get(), {std::weak_ptr<Tensor>{tensor}, id}});
return id; return id;
} }
} }


+ 2
- 2
imperative/src/include/megbrain/imperative/profiler.h View File

@@ -40,8 +40,8 @@ struct ProfileEntry {
std::vector<std::tuple<CompNode, TimeClosure, TimeClosure>> device_list; std::vector<std::tuple<CompNode, TimeClosure, TimeClosure>> device_list;
std::vector<ProfileTensor> inputs; std::vector<ProfileTensor> inputs;
std::vector<ProfileTensor> outputs; std::vector<ProfileTensor> outputs;
ssize_t memory = 0;
ssize_t computation = 0;
long long memory = 0;
long long computation = 0;
#if MGB_ENABLE_JSON #if MGB_ENABLE_JSON
std::shared_ptr<json::Value> param; std::shared_ptr<json::Value> param;
#endif #endif


Loading…
Cancel
Save