diff --git a/imperative/src/impl/profiler.cpp b/imperative/src/impl/profiler.cpp index 681cccbe..ccb98081 100644 --- a/imperative/src/impl/profiler.cpp +++ b/imperative/src/impl/profiler.cpp @@ -83,7 +83,8 @@ size_t TensorRecorder::record_tensor(const TensorPtr& tensor) { return id; } else { 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}, id}}); return id; } } diff --git a/imperative/src/include/megbrain/imperative/profiler.h b/imperative/src/include/megbrain/imperative/profiler.h index d1cd37de..b2452242 100644 --- a/imperative/src/include/megbrain/imperative/profiler.h +++ b/imperative/src/include/megbrain/imperative/profiler.h @@ -40,8 +40,8 @@ struct ProfileEntry { std::vector> device_list; std::vector inputs; std::vector outputs; - ssize_t memory = 0; - ssize_t computation = 0; + long long memory = 0; + long long computation = 0; #if MGB_ENABLE_JSON std::shared_ptr param; #endif