diff --git a/imperative/src/impl/profiler.cpp b/imperative/src/impl/profiler.cpp index 681cccbea3715c5303995e11f90d39edc3169aae..ccb98081d790038f9d6b01ea7cfa3ddb1636c481 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 d1cd37deb65d62f407782dc666864d65870fb342..b24522420e5758cef4a903a7401181d795b053a2 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