File: builtin-expect-with-probability.cpp

package info (click to toggle)
llvm-toolchain-13 1%3A13.0.1-11
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,418,840 kB
  • sloc: cpp: 5,290,826; ansic: 996,570; asm: 544,593; python: 188,212; objc: 72,027; lisp: 30,291; f90: 25,395; sh: 24,898; javascript: 9,780; pascal: 9,398; perl: 7,484; ml: 5,432; awk: 3,523; makefile: 2,913; xml: 953; cs: 573; fortran: 539
file content (101 lines) | stat: -rw-r--r-- 2,530 bytes parent folder | download | duplicates (20)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// RUN: %clang_cc1 -triple x86_64-unknown-unknown -emit-llvm -o - %s -O1 -disable-llvm-passes | FileCheck %s --check-prefix=ALL --check-prefix=O1
// RUN: %clang_cc1 -triple x86_64-unknown-unknown -emit-llvm -o - %s -O0 | FileCheck %s --check-prefix=ALL --check-prefix=O0
extern int global;

int expect_taken(int x) {
// ALL-LABEL: expect_taken
// O1:        call i64 @llvm.expect.with.probability.i64(i64 {{%.*}}, i64 1, double 9.000000e-01)
// O0-NOT:    @llvm.expect.with.probability

  if (__builtin_expect_with_probability(x == 100, 1, 0.9)) {
    return 0;
  }
  return x;
}

int expect_not_taken(int x) {
// ALL-LABEL: expect_not_taken
// O1:        call i64 @llvm.expect.with.probability.i64(i64 {{%.*}}, i64 0, double 9.000000e-01)
// O0-NOT:    @llvm.expect.with.probability

  if (__builtin_expect_with_probability(x == 100, 0, 0.9)) {
    return 0;
  }
  return x;
}

struct S {
  static constexpr int prob = 1;
};

template<typename T>
int expect_taken_template(int x) {
// ALL-LABEL: expect_taken_template
// O1:        call i64 @llvm.expect.with.probability.i64(i64 {{%.*}}, i64 1, double 1.000000e+00)
// O0-NOT:    @llvm.expect.with.probability

	if (__builtin_expect_with_probability (x == 100, 1, T::prob)) {
		return 0;
	}
	return x;
}

int f() {
  return expect_taken_template<S>(global);
}

int x;
extern "C" {
  int y(void);
}
void foo();

void expect_value_side_effects() {
// ALL-LABEL: expect_value_side_effects
// ALL: [[CALL:%.*]] = call i32 @y
// O1:  [[SEXT:%.*]] = sext i32 [[CALL]] to i64
// O1:  call i64 @llvm.expect.with.probability.i64(i64 {{%.*}}, i64 [[SEXT]], double 6.000000e-01)
// O0-NOT: @llvm.expect.with.probability

  if (__builtin_expect_with_probability(x, y(), 0.6))
    foo();
}

int switch_cond(int x) {
// ALL-LABEL: switch_cond
// O1:        call i64 @llvm.expect.with.probability.i64(i64 {{%.*}}, i64 1, double 8.000000e-01)
// O0-NOT:    @llvm.expect.with.probability

  switch (__builtin_expect_with_probability(x, 1, 0.8)) {
  case 0:
    x = x + 0;
  case 1:
    x = x + 1;
  case 2:
    x = x + 2;
  case 5:
    x = x + 5;
  default:
    x = x + 6;
  }
  return x;
}

constexpr double prob = 0.8;

int variable_expected(int stuff) {
// ALL-LABEL: variable_expected
// O1: call i64 @llvm.expect.with.probability.i64(i64 {{%.*}}, i64 {{%.*}}, double 8.000000e-01)
// O0-NOT: @llvm.expect.with.probability

  int res = 0;

  switch(__builtin_expect_with_probability(stuff, stuff, prob)) {
    case 0:
      res = 1;
      break;
    default:
      break;
  }
  return res;
}