xref: /llvm-project/llvm/test/tools/llvm-profgen/cold-profile-trimming.test (revision 484a569eea7ba294d84b9a700e6fcc2f97626320)
1; RUN: llvm-profgen --format=text --unsymbolized-profile=%S/Inputs/cold-profile-trimming.raw.prof --binary=%S/Inputs/inline-noprobe2.perfbin --output=%t1 --use-offset=0 --trim-cold-profile=0
2; RUN: FileCheck %s --input-file %t1 --check-prefix=CHECK-NOTRIM
3; RUN: llvm-profgen --format=text --unsymbolized-profile=%S/Inputs/cold-profile-trimming.raw.prof --binary=%S/Inputs/inline-noprobe2.perfbin --output=%t1 --use-offset=0 --trim-cold-profile=1 --profile-summary-cold-count=1000
4; RUN: FileCheck %s --input-file %t1 --check-prefix=CHECK-TRIM
5
6;CHECK-NOTRIM: partition_pivot_last:5187:7
7;CHECK-NOTRIM: partition_pivot_first:3010:5
8;CHECK-NOTRIM: quick_sort:903:25
9;CHECK-NOTRIM: main:820:0
10
11;CHECK-TRIM: partition_pivot_last:5187:7
12;CHECK-TRIM: partition_pivot_first:3010:5
13;CHECK-TRIM-NOT: quick_sort:903:25
14;CHECK-TRIM-NOT: main:820:0
15
16; original code:
17; clang -O3 -g -fno-optimize-sibling-calls -fdebug-info-for-profiling qsort.c -o a.out
18#include <stdio.h>
19#include <stdlib.h>
20
21void swap(int *a, int *b) {
22	int t = *a;
23	*a = *b;
24	*b = t;
25}
26
27int partition_pivot_last(int* array, int low, int high) {
28	int pivot = array[high];
29	int i = low - 1;
30	for (int j = low; j < high; j++)
31		if (array[j] < pivot)
32			swap(&array[++i], &array[j]);
33	swap(&array[i + 1], &array[high]);
34	return (i + 1);
35}
36
37int partition_pivot_first(int* array, int low, int high) {
38	int pivot = array[low];
39	int i = low + 1;
40	for (int j = low + 1; j <= high; j++)
41		if (array[j] < pivot) { if (j != i) swap(&array[i], &array[j]); i++;}
42	swap(&array[i - 1], &array[low]);
43	return i - 1;
44}
45
46void quick_sort(int* array, int low, int high, int (*partition_func)(int *, int, int)) {
47	if (low < high) {
48		int pi = (*partition_func)(array, low, high);
49		quick_sort(array, low, pi - 1, partition_func);
50		quick_sort(array, pi + 1, high, partition_func);
51	}
52}
53
54int main() {
55	const int size = 200;
56	int sum = 0;
57	int *array = malloc(size * sizeof(int));
58	for(int i = 0; i < 100 * 1000; i++) {
59		for(int j = 0; j < size; j++)
60			array[j] = j % 10 ? rand() % size: j;
61		int (*fptr)(int *, int, int) = i % 3 ? partition_pivot_last : partition_pivot_first;
62		quick_sort(array, 0, size - 1, fptr);
63		sum += array[i % size];
64	}
65	printf("sum=%d\n", sum);
66
67	return 0;
68}
69