llvm/test/Transforms/SampleProfile/use-sample-profile-attr.ll

   1 ; RUN: opt < %s -passes=sample-profile -sample-profile-file=%S/Inputs/use-sample-profile-attr.prof -S | FileCheck %s
   2
   3 @.str = private unnamed_addr constant [11 x i8] c"sum is %d\0A\00", align 1
   4
   5 ;; @goo doesn't have use-sample-profile attribute so it won't have
   6 ;; profile annotated.
   7 ; CHECK-NOT: @goo{{.*}} !prof
   8 define void @goo() !dbg !26 {
   9   ret void
  10 }
  11
  12 ;; @foo has use-sample-profile attribute so it will have profile annotated.
  13 ; CHECK: @foo{{.*}} !prof ![[HDRCNT1:[0-9]+]]
  14 define i32 @foo(i32 %x, i32 %y) #0 !dbg !4 {
  15 entry:
  16   %x.addr = alloca i32, align 4
  17   %y.addr = alloca i32, align 4
  18   store i32 %x, ptr %x.addr, align 4
  19   store i32 %y, ptr %y.addr, align 4
  20   %0 = load i32, ptr %x.addr, align 4, !dbg !11
  21   %1 = load i32, ptr %y.addr, align 4, !dbg !11
  22   %add = add nsw i32 %0, %1, !dbg !11
  23   ret i32 %add, !dbg !11
  24 }
  25
  26 ;; @main has use-sample-profile attribute so it will have profile annotated.
  27 ; CHECK: @main{{.*}} !prof ![[HDRCNT2:[0-9]+]]
  28 define i32 @main() #1 !dbg !7 {
  29 entry:
  30   %retval = alloca i32, align 4
  31   %s = alloca i32, align 4
  32   %i = alloca i32, align 4
  33   store i32 0, ptr %retval
  34   store i32 0, ptr %i, align 4, !dbg !12
  35   br label %while.cond, !dbg !13
  36
  37 while.cond:                                       ; preds = %if.end, %entry
  38   %0 = load i32, ptr %i, align 4, !dbg !14
  39   %inc = add nsw i32 %0, 1, !dbg !14
  40   store i32 %inc, ptr %i, align 4, !dbg !14
  41   %cmp = icmp slt i32 %0, 400000000, !dbg !14
  42   br i1 %cmp, label %while.body, label %while.end, !dbg !14
  43
  44 while.body:                                       ; preds = %while.cond
  45   %1 = load i32, ptr %i, align 4, !dbg !16
  46   %cmp1 = icmp ne i32 %1, 100, !dbg !16
  47   br i1 %cmp1, label %if.then, label %if.else, !dbg !16
  48
  49
  50 if.then:                                          ; preds = %while.body
  51   %2 = load i32, ptr %i, align 4, !dbg !18
  52   %3 = load i32, ptr %s, align 4, !dbg !18
  53 ;; @foo is inlined because the callsite is hot and @foo has use-sample-profile
  54 ;; attribute.
  55 ; CHECK: if.then:
  56 ; CHECK-NOT: call i32 @foo
  57   %call = call i32 @foo(i32 %2, i32 %3), !dbg !18
  58   store i32 %call, ptr %s, align 4, !dbg !18
  59   br label %if.end, !dbg !18
  60
  61 if.else:                                          ; preds = %while.body
  62 ;; @goo is not inlined because @goo doesn't have use-sample-profile attribute.
  63 ; CHECK: if.else:
  64 ; CHECK: call void @goo
  65   call void @goo(), !dbg !27
  66   store i32 30, ptr %s, align 4, !dbg !20
  67   br label %if.end
  68
  69 if.end:                                           ; preds = %if.else, %if.then
  70   br label %while.cond, !dbg !22
  71
  72 while.end:                                        ; preds = %while.cond
  73   %4 = load i32, ptr %s, align 4, !dbg !24
  74   %call2 = call i32 (ptr, ...) @printf(ptr @.str, i32 %4), !dbg !24
  75   ret i32 0, !dbg !25
  76 }
  77
  78 ; CHECK: ![[HDRCNT1]] = !{!"function_entry_count", i64 11}
  79 ; CHECK: ![[HDRCNT2]] = !{!"function_entry_count", i64 2}
  80
  81 attributes #0 = {"use-sample-profile"}
  82 attributes #1 = {"use-sample-profile"}
  83
  84 declare i32 @printf(ptr, ...)
  85
  86 !llvm.dbg.cu = !{!0}
  87 !llvm.module.flags = !{!8, !9}
  88 !llvm.ident = !{!10}
  89
  90 !0 = distinct !DICompileUnit(language: DW_LANG_C_plus_plus, producer: "clang version 3.5 ", isOptimized: false, emissionKind: NoDebug, file: !1, enums: !2, retainedTypes: !2, globals: !2, imports: !2)
  91 !1 = !DIFile(filename: "calls.cc", directory: ".")
  92 !2 = !{}
  93 !4 = distinct !DISubprogram(name: "foo", line: 3, isLocal: false, isDefinition: true, virtualIndex: 6, flags: DIFlagPrototyped, isOptimized: false, unit: !0, scopeLine: 3, file: !1, scope: !5, type: !6, retainedNodes: !2)
  94 !5 = !DIFile(filename: "calls.cc", directory: ".")
  95 !6 = !DISubroutineType(types: !2)
  96 !7 = distinct !DISubprogram(name: "main", line: 7, isLocal: false, isDefinition: true, virtualIndex: 6, flags: DIFlagPrototyped, isOptimized: false, unit: !0, scopeLine: 7, file: !1, scope: !5, type: !6, retainedNodes: !2)
  97 !8 = !{i32 2, !"Dwarf Version", i32 4}
  98 !9 = !{i32 1, !"Debug Info Version", i32 3}
  99 !10 = !{!"clang version 3.5 "}
 100 !11 = !DILocation(line: 4, scope: !4)
 101 !12 = !DILocation(line: 8, scope: !7)
 102 !13 = !DILocation(line: 9, scope: !7)
 103 !14 = !DILocation(line: 9, scope: !15)
 104 !15 = !DILexicalBlockFile(discriminator: 2, file: !1, scope: !7)
 105 !16 = !DILocation(line: 10, scope: !17)
 106 !17 = distinct !DILexicalBlock(line: 10, column: 0, file: !1, scope: !7)
 107 !18 = !DILocation(line: 10, scope: !19)
 108 !19 = !DILexicalBlockFile(discriminator: 2, file: !1, scope: !17)
 109 !20 = !DILocation(line: 10, scope: !21)
 110 !21 = !DILexicalBlockFile(discriminator: 4, file: !1, scope: !17)
 111 !22 = !DILocation(line: 10, scope: !23)
 112 !23 = !DILexicalBlockFile(discriminator: 6, file: !1, scope: !17)
 113 !24 = !DILocation(line: 11, scope: !7)
 114 !25 = !DILocation(line: 12, scope: !7)
 115 !26 = distinct !DISubprogram(name: "goo", line: 11, isLocal: false, isDefinition: true, virtualIndex: 6, flags: DIFlagPrototyped, isOptimized: false, unit: !0, scopeLine: 3, file: !1, scope: !5, type: !6, retainedNodes: !2)
 116 !27 = !DILocation(line: 11, column: 20, scope: !7)