OLD | NEW |
1 // Copyright 2009 The RE2 Authors. All Rights Reserved. | 1 // Copyright 2009 The RE2 Authors. All Rights Reserved. |
2 // Use of this source code is governed by a BSD-style | 2 // Use of this source code is governed by a BSD-style |
3 // license that can be found in the LICENSE file. | 3 // license that can be found in the LICENSE file. |
4 | 4 |
5 #include "util/util.h" | 5 #include "util/util.h" |
6 #include "util/flags.h" | 6 #include "util/flags.h" |
7 #include "util/benchmark.h" | 7 #include "util/benchmark.h" |
8 #include "re2/re2.h" | 8 #include "re2/re2.h" |
9 | 9 |
10 DEFINE_string(test_tmpdir, "/var/tmp", "temp directory"); | 10 DEFINE_string(test_tmpdir, "/var/tmp", "temp directory"); |
11 | 11 |
12 using testing::Benchmark; | 12 using testing::Benchmark; |
13 using namespace re2; | 13 using namespace re2; |
14 | 14 |
15 static Benchmark* benchmarks[10000]; | 15 static Benchmark* benchmarks[10000]; |
16 static int nbenchmarks; | 16 static int nbenchmarks; |
17 | 17 |
18 void Benchmark::Register() { | 18 void Benchmark::Register() { |
19 benchmarks[nbenchmarks] = this; | 19 benchmarks[nbenchmarks] = this; |
20 if(lo < 1) | 20 if(lo < 1) |
21 lo = 1; | 21 lo = 1; |
22 if(hi < lo) | 22 if(hi < lo) |
23 hi = lo; | 23 hi = lo; |
24 nbenchmarks++; | 24 nbenchmarks++; |
25 } | 25 } |
26 | 26 |
27 static int64 nsec() { | 27 static int64 nsec() { |
28 #if defined(__APPLE__) | |
29 struct timeval tv; | 28 struct timeval tv; |
30 if(gettimeofday(&tv, 0) < 0) | 29 if(gettimeofday(&tv, 0) < 0) |
31 return -1; | 30 return -1; |
32 return (int64)tv.tv_sec*1000*1000*1000 + tv.tv_usec*1000; | 31 return (int64)tv.tv_sec*1000*1000*1000 + tv.tv_usec*1000; |
33 #elif defined(_WIN32) | |
34 // https://msdn.microsoft.com/en-us/library/windows/desktop/dn553408.asp
x | |
35 // describes how to query ticks and convert to microseconds. Of course, | |
36 // what we want in this case are nanoseconds. Also, note that .QuadPart | |
37 // is a signed 64-bit integer, so casting to int64 shouldn't be needed. | |
38 LARGE_INTEGER freq; | |
39 QueryPerformanceFrequency(&freq); | |
40 LARGE_INTEGER ticks; | |
41 QueryPerformanceCounter(&ticks); | |
42 ticks.QuadPart *= 1000*1000*1000; | |
43 ticks.QuadPart /= freq.QuadPart; | |
44 return ticks.QuadPart; | |
45 #else | |
46 struct timespec tp; | |
47 if(clock_gettime(CLOCK_REALTIME, &tp) < 0) | |
48 return -1; | |
49 return (int64)tp.tv_sec*1000*1000*1000 + tp.tv_nsec; | |
50 #endif | |
51 } | 32 } |
52 | 33 |
53 static int64 bytes; | 34 static int64 bytes; |
54 static int64 ns; | 35 static int64 ns; |
55 static int64 t0; | 36 static int64 t0; |
56 static int64 items; | 37 static int64 items; |
57 | 38 |
58 void SetBenchmarkBytesProcessed(long long x) { | 39 void SetBenchmarkBytesProcessed(long long x) { |
59 bytes = x; | 40 bytes = x; |
60 } | 41 } |
(...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 // TODO(rsc): Threaded benchmarks. | 98 // TODO(rsc): Threaded benchmarks. |
118 if(nthread != 1) | 99 if(nthread != 1) |
119 return; | 100 return; |
120 | 101 |
121 // run once in case it's expensive | 102 // run once in case it's expensive |
122 n = 1; | 103 n = 1; |
123 runN(b, n, siz); | 104 runN(b, n, siz); |
124 while(ns < (int)1e9 && n < (int)1e9) { | 105 while(ns < (int)1e9 && n < (int)1e9) { |
125 last = n; | 106 last = n; |
126 if(ns/n == 0) | 107 if(ns/n == 0) |
127 » » » n = (int)1e9; | 108 » » » n = 1e9; |
128 else | 109 else |
129 » » » n = (int)1e9 / static_cast<int>(ns/n); | 110 » » » n = 1e9 / (ns/n); |
130 | 111 |
131 n = max(last+1, min(n+n/2, 100*last)); | 112 n = max(last+1, min(n+n/2, 100*last)); |
132 n = round(n); | 113 n = round(n); |
133 runN(b, n, siz); | 114 runN(b, n, siz); |
134 } | 115 } |
135 | 116 |
136 char mb[100]; | 117 char mb[100]; |
137 char suf[100]; | 118 char suf[100]; |
138 mb[0] = '\0'; | 119 mb[0] = '\0'; |
139 suf[0] = '\0'; | 120 suf[0] = '\0'; |
(...skipping 23 matching lines...) Expand all Loading... |
163 int main(int argc, const char** argv) { | 144 int main(int argc, const char** argv) { |
164 for(int i = 0; i < nbenchmarks; i++) { | 145 for(int i = 0; i < nbenchmarks; i++) { |
165 Benchmark* b = benchmarks[i]; | 146 Benchmark* b = benchmarks[i]; |
166 if(match(b->name, argc, argv)) | 147 if(match(b->name, argc, argv)) |
167 for(int j = b->threadlo; j <= b->threadhi; j++) | 148 for(int j = b->threadlo; j <= b->threadhi; j++) |
168 for(int k = max(b->lo, 1); k <= max(b->hi, 1); k
<<=1) | 149 for(int k = max(b->lo, 1); k <= max(b->hi, 1); k
<<=1) |
169 RunBench(b, j, k); | 150 RunBench(b, j, k); |
170 } | 151 } |
171 } | 152 } |
172 | 153 |
OLD | NEW |