llvm/test/Transforms/FunctionAttrs/convergent.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-attributes
   2 ; RUN: opt -passes=function-attrs -S < %s | FileCheck %s
   3
   4 define i32 @nonleaf() convergent {
   5 ; CHECK: Function Attrs: mustprogress nofree norecurse nosync nounwind willreturn memory(none)
   6 ; CHECK-LABEL: define {{[^@]+}}@nonleaf
   7 ; CHECK-SAME: () #[[ATTR0:[0-9]+]] {
   8 ; CHECK-NEXT:    [[A:%.*]] = call i32 @leaf()
   9 ; CHECK-NEXT:    ret i32 [[A]]
  10 ;
  11   %a = call i32 @leaf()
  12   ret i32 %a
  13 }
  14
  15 define i32 @leaf() convergent {
  16 ; CHECK: Function Attrs: mustprogress nofree norecurse nosync nounwind willreturn memory(none)
  17 ; CHECK-LABEL: define {{[^@]+}}@leaf
  18 ; CHECK-SAME: () #[[ATTR0]] {
  19 ; CHECK-NEXT:    ret i32 0
  20 ;
  21   ret i32 0
  22 }
  23
  24 declare i32 @k() convergent
  25
  26 define i32 @extern() convergent {
  27 ; CHECK: Function Attrs: convergent
  28 ; CHECK-LABEL: define {{[^@]+}}@extern
  29 ; CHECK-SAME: () #[[ATTR1:[0-9]+]] {
  30 ; CHECK-NEXT:    [[A:%.*]] = call i32 @k() #[[ATTR1]]
  31 ; CHECK-NEXT:    ret i32 [[A]]
  32 ;
  33   %a = call i32 @k() convergent
  34   ret i32 %a
  35 }
  36
  37 ; Convergent should not be removed on the function here.  Although the call is
  38 ; not explicitly convergent, it picks up the convergent attr from the callee.
  39 define i32 @extern_non_convergent_call() convergent {
  40 ; CHECK: Function Attrs: convergent
  41 ; CHECK-LABEL: define {{[^@]+}}@extern_non_convergent_call
  42 ; CHECK-SAME: () #[[ATTR1]] {
  43 ; CHECK-NEXT:    [[A:%.*]] = call i32 @k()
  44 ; CHECK-NEXT:    ret i32 [[A]]
  45 ;
  46   %a = call i32 @k()
  47   ret i32 %a
  48 }
  49
  50 define i32 @indirect_convergent_call(ptr %f) convergent {
  51 ; CHECK: Function Attrs: convergent
  52 ; CHECK-LABEL: define {{[^@]+}}@indirect_convergent_call
  53 ; CHECK-SAME: (ptr nocapture readonly [[F:%.*]]) #[[ATTR1]] {
  54 ; CHECK-NEXT:    [[A:%.*]] = call i32 [[F]]() #[[ATTR1]]
  55 ; CHECK-NEXT:    ret i32 [[A]]
  56 ;
  57   %a = call i32 %f() convergent
  58   ret i32 %a
  59 }
  60 ; Give indirect_non_convergent_call the norecurse attribute so we get a
  61 ; "Function Attrs" comment in the output.
  62 define i32 @indirect_non_convergent_call(ptr %f) convergent norecurse {
  63 ; CHECK: Function Attrs: norecurse
  64 ; CHECK-LABEL: define {{[^@]+}}@indirect_non_convergent_call
  65 ; CHECK-SAME: (ptr nocapture readonly [[F:%.*]]) #[[ATTR2:[0-9]+]] {
  66 ; CHECK-NEXT:    [[A:%.*]] = call i32 [[F]]()
  67 ; CHECK-NEXT:    ret i32 [[A]]
  68 ;
  69   %a = call i32 %f()
  70   ret i32 %a
  71 }
  72
  73 declare void @llvm.nvvm.barrier0() convergent
  74
  75 define i32 @intrinsic() convergent {
  76   ; Implicitly convergent, because the intrinsic is convergent.
  77 ; CHECK: Function Attrs: convergent norecurse nounwind
  78 ; CHECK-LABEL: define {{[^@]+}}@intrinsic
  79 ; CHECK-SAME: () #[[ATTR4:[0-9]+]] {
  80 ; CHECK-NEXT:    call void @llvm.nvvm.barrier0()
  81 ; CHECK-NEXT:    ret i32 0
  82 ;
  83   call void @llvm.nvvm.barrier0()
  84   ret i32 0
  85 }
  86
  87 define i32 @recursive1() convergent {
  88 ; CHECK: Function Attrs: nofree nosync nounwind memory(none)
  89 ; CHECK-LABEL: define {{[^@]+}}@recursive1
  90 ; CHECK-SAME: () #[[ATTR5:[0-9]+]] {
  91 ; CHECK-NEXT:    [[A:%.*]] = call i32 @recursive2() #[[ATTR1]]
  92 ; CHECK-NEXT:    ret i32 [[A]]
  93 ;
  94   %a = call i32 @recursive2() convergent
  95   ret i32 %a
  96 }
  97
  98 define i32 @recursive2() convergent {
  99 ; CHECK: Function Attrs: nofree nosync nounwind memory(none)
 100 ; CHECK-LABEL: define {{[^@]+}}@recursive2
 101 ; CHECK-SAME: () #[[ATTR5]] {
 102 ; CHECK-NEXT:    [[A:%.*]] = call i32 @recursive1() #[[ATTR1]]
 103 ; CHECK-NEXT:    ret i32 [[A]]
 104 ;
 105   %a = call i32 @recursive1() convergent
 106   ret i32 %a
 107 }
 108
 109 define i32 @noopt() convergent optnone noinline {
 110 ; CHECK: Function Attrs: convergent noinline optnone
 111 ; CHECK-LABEL: define {{[^@]+}}@noopt
 112 ; CHECK-SAME: () #[[ATTR6:[0-9]+]] {
 113 ; CHECK-NEXT:    [[A:%.*]] = call i32 @noopt_friend() #[[ATTR1]]
 114 ; CHECK-NEXT:    ret i32 0
 115 ;
 116   %a = call i32 @noopt_friend() convergent
 117   ret i32 0
 118 }
 119
 120 ; A function which is mutually-recursive with a convergent, optnone function
 121 ; shouldn't have its convergent attribute stripped.
 122 define i32 @noopt_friend() convergent {
 123 ; CHECK: Function Attrs: convergent
 124 ; CHECK-LABEL: define {{[^@]+}}@noopt_friend
 125 ; CHECK-SAME: () #[[ATTR1]] {
 126 ; CHECK-NEXT:    [[A:%.*]] = call i32 @noopt()
 127 ; CHECK-NEXT:    ret i32 0
 128 ;
 129   %a = call i32 @noopt()
 130   ret i32 0
 131 }