third-party/benchmark/test/state_assembly_test.cc

   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(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(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 }