1 | #include <benchmark/benchmark.h> |
2 | |
3 | #ifdef __clang__ |
4 | #pragma clang diagnostic ignored "-Wreturn-type" |
5 | #endif |
6 | |
7 | // clang-format off |
8 | extern "C" { |
9 | extern int ExternInt; |
10 | benchmark::State& GetState(); |
11 | void Fn(); |
12 | } |
13 | // clang-format on |
14 | |
15 | using benchmark::State; |
16 | |
17 | // CHECK-LABEL: test_for_auto_loop: |
18 | extern "C" int test_for_auto_loop() { |
19 | State& S = GetState(); |
20 | int x = 42; |
21 | // CHECK: [[CALL:call(q)*]] _ZN9benchmark5State16StartKeepRunningEv |
22 | // CHECK-NEXT: testq %rbx, %rbx |
23 | // CHECK-NEXT: je [[LOOP_END:.*]] |
24 | |
25 | for (auto _ : S) { |
26 | // CHECK: .L[[LOOP_HEAD:[a-zA-Z0-9_]+]]: |
27 | // CHECK-GNU-NEXT: subq $1, %rbx |
28 | // CHECK-CLANG-NEXT: {{(addq \$1, %rax|incq %rax|addq \$-1, %rbx)}} |
29 | // CHECK-NEXT: jne .L[[LOOP_HEAD]] |
30 | benchmark::DoNotOptimize(value&: x); |
31 | } |
32 | // CHECK: [[LOOP_END]]: |
33 | // CHECK: [[CALL]] _ZN9benchmark5State17FinishKeepRunningEv |
34 | |
35 | // CHECK: movl $101, %eax |
36 | // CHECK: ret |
37 | return 101; |
38 | } |
39 | |
40 | // CHECK-LABEL: test_while_loop: |
41 | extern "C" int test_while_loop() { |
42 | State& S = GetState(); |
43 | int x = 42; |
44 | |
45 | // CHECK: j{{(e|mp)}} .L[[LOOP_HEADER:[a-zA-Z0-9_]+]] |
46 | // CHECK-NEXT: .L[[LOOP_BODY:[a-zA-Z0-9_]+]]: |
47 | while (S.KeepRunning()) { |
48 | // CHECK-GNU-NEXT: subq $1, %[[IREG:[a-z]+]] |
49 | // CHECK-CLANG-NEXT: {{(addq \$-1,|decq)}} %[[IREG:[a-z]+]] |
50 | // CHECK: movq %[[IREG]], [[DEST:.*]] |
51 | benchmark::DoNotOptimize(value&: x); |
52 | } |
53 | // CHECK-DAG: movq [[DEST]], %[[IREG]] |
54 | // CHECK-DAG: testq %[[IREG]], %[[IREG]] |
55 | // CHECK-DAG: jne .L[[LOOP_BODY]] |
56 | // CHECK-DAG: .L[[LOOP_HEADER]]: |
57 | |
58 | // CHECK: cmpb $0 |
59 | // CHECK-NEXT: jne .L[[LOOP_END:[a-zA-Z0-9_]+]] |
60 | // CHECK: [[CALL:call(q)*]] _ZN9benchmark5State16StartKeepRunningEv |
61 | |
62 | // CHECK: .L[[LOOP_END]]: |
63 | // CHECK: [[CALL]] _ZN9benchmark5State17FinishKeepRunningEv |
64 | |
65 | // CHECK: movl $101, %eax |
66 | // CHECK: ret |
67 | return 101; |
68 | } |
69 | |