GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the tensor benchmarks on apple devices

Browse Source
This commit is contained in:
Benoit Steiner 2016-01-28 21:08:07 -08:00
parent 10bea90c4a
commit e4f83bae5d
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
bench/tensors/benchmark_main.cc
View File

@ -49,12 +49,27 @@ static int Round(int n) {
} }
return 10*base; return 10*base;
} }
#ifdef __APPLE__
#include <mach/mach_time.h>
static mach_timebase_info_data_t g_time_info;
static void __attribute__((constructor)) init_info() {
mach_timebase_info(&g_time_info);
}
#endif
static int64_t NanoTime() { static int64_t NanoTime() {
#if defined(__APPLE__)
uint64_t t = mach_absolute_time();
return t * g_time_info.numer / g_time_info.denom;
#else
struct timespec t; struct timespec t;
t.tv_sec = t.tv_nsec = 0; t.tv_sec = t.tv_nsec = 0;
clock_gettime(CLOCK_MONOTONIC, &t); clock_gettime(CLOCK_MONOTONIC, &t);
return static_cast<int64_t>(t.tv_sec) * 1000000000LL + t.tv_nsec; return static_cast<int64_t>(t.tv_sec) * 1000000000LL + t.tv_nsec;
#endif
} }
namespace testing { namespace testing {
Benchmark* Benchmark::Arg(int arg) { Benchmark* Benchmark::Arg(int arg) {
args_.push_back(arg); args_.push_back(arg);