mirror of https://github.com/alibaba/MNN.git
72 lines
1.7 KiB
C++
72 lines
1.7 KiB
C++
//
|
|
// AutoTime.cpp
|
|
// MNN
|
|
//
|
|
// Created by MNN on 2018/07/14.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#if defined(_MSC_VER)
|
|
#include <Windows.h>
|
|
#else
|
|
#include <sys/time.h>
|
|
#endif
|
|
#include <MNN/AutoTime.hpp>
|
|
#include "core/Macro.h"
|
|
|
|
namespace MNN {
|
|
|
|
Timer::Timer() {
|
|
reset();
|
|
}
|
|
|
|
Timer::~Timer() {
|
|
// do nothing
|
|
}
|
|
|
|
void Timer::reset() {
|
|
#if defined(_MSC_VER)
|
|
LARGE_INTEGER time, freq;
|
|
QueryPerformanceFrequency(&freq);
|
|
QueryPerformanceCounter(&time);
|
|
uint64_t sec = time.QuadPart / freq.QuadPart;
|
|
uint64_t usec = (time.QuadPart % freq.QuadPart) * 1000000 / freq.QuadPart;
|
|
mLastResetTime = sec * 1000000 + usec;
|
|
#else
|
|
struct timeval Current;
|
|
gettimeofday(&Current, nullptr);
|
|
mLastResetTime = Current.tv_sec * 1000000 + Current.tv_usec;
|
|
#endif
|
|
}
|
|
|
|
uint64_t Timer::durationInUs() {
|
|
#if defined(_MSC_VER)
|
|
LARGE_INTEGER time, freq;
|
|
QueryPerformanceCounter(&time);
|
|
QueryPerformanceFrequency(&freq);
|
|
uint64_t sec = time.QuadPart / freq.QuadPart;
|
|
uint64_t usec = (time.QuadPart % freq.QuadPart) * 1000000 / freq.QuadPart;
|
|
auto lastTime = sec * 1000000 + usec;
|
|
#else
|
|
struct timeval Current;
|
|
gettimeofday(&Current, nullptr);
|
|
auto lastTime = Current.tv_sec * 1000000 + Current.tv_usec;
|
|
#endif
|
|
|
|
return lastTime - mLastResetTime;
|
|
}
|
|
|
|
AutoTime::AutoTime(int line, const char* func) : Timer() {
|
|
mName = ::strdup(func);
|
|
mLine = line;
|
|
}
|
|
AutoTime::~AutoTime() {
|
|
auto timeInUs = durationInUs();
|
|
MNN_PRINT("%s, %d, cost time: %f ms\n", mName, mLine, (float)timeInUs / 1000.0f);
|
|
free(mName);
|
|
}
|
|
|
|
} // namespace MNN
|