Commit eebf0dd7 authored by catree's avatar catree

Fix integer overflow when accumulating timing values.

parent e75576e1
...@@ -3074,7 +3074,7 @@ void Net::setHalideScheduler(const String& scheduler) ...@@ -3074,7 +3074,7 @@ void Net::setHalideScheduler(const String& scheduler)
int64 Net::getPerfProfile(std::vector<double>& timings) int64 Net::getPerfProfile(std::vector<double>& timings)
{ {
timings = std::vector<double>(impl->layersTimings.begin() + 1, impl->layersTimings.end()); timings = std::vector<double>(impl->layersTimings.begin() + 1, impl->layersTimings.end());
int64 total = std::accumulate(timings.begin(), timings.end(), 0); int64 total = (int64)std::accumulate(timings.begin(), timings.end(), 0.0);
return total; return total;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment