llvm/test/CodeGen/NVPTX/nvvm-reflect-arch-O0.ll

   1 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_52 -mattr=+ptx64 -O0 | FileCheck %s --check-prefix=SM_52
   2 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_70 -mattr=+ptx64 -O0 | FileCheck %s --check-prefix=SM_70
   3 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_90 -mattr=+ptx72 -O0 | FileCheck %s --check-prefix=SM_90
   4
   5 @.str = private unnamed_addr constant [12 x i8] c"__CUDA_ARCH\00"
   6 @.str1 = constant [11 x i8] c"__CUDA_FTZ\00"
   7
   8 declare i32 @__nvvm_reflect(ptr)
   9
  10 ;      SM_52: .visible .func  (.param .b32 func_retval0) foo()
  11 ;      SM_52: mov.b32         %[[REG:.+]], 3;
  12 ; SM_52-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
  13 ; SM_52-NEXT: ret;
  14 ;
  15 ;      SM_70: .visible .func  (.param .b32 func_retval0) foo()
  16 ;      SM_70: mov.b32         %[[REG:.+]], 2;
  17 ; SM_70-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
  18 ; SM_70-NEXT: ret;
  19 ;
  20 ;      SM_90: .visible .func  (.param .b32 func_retval0) foo()
  21 ;      SM_90: mov.b32         %[[REG:.+]], 1;
  22 ; SM_90-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
  23 ; SM_90-NEXT: ret;
  24 define i32 @foo() {
  25 entry:
  26   %call = call i32 @__nvvm_reflect(ptr @.str)
  27   %cmp = icmp uge i32 %call, 900
  28   br i1 %cmp, label %if.then, label %if.else
  29
  30 if.then:
  31   br label %return
  32
  33 if.else:
  34   %call1 = call i32 @__nvvm_reflect(ptr @.str)
  35   %cmp2 = icmp uge i32 %call1, 700
  36   br i1 %cmp2, label %if.then3, label %if.else4
  37
  38 if.then3:
  39   br label %return
  40
  41 if.else4:
  42   %call5 = call i32 @__nvvm_reflect(ptr @.str)
  43   %cmp6 = icmp uge i32 %call5, 520
  44   br i1 %cmp6, label %if.then7, label %if.else8
  45
  46 if.then7:
  47   br label %return
  48
  49 if.else8:
  50   br label %return
  51
  52 return:
  53   %retval.0 = phi i32 [ 1, %if.then ], [ 2, %if.then3 ], [ 3, %if.then7 ], [ 4, %if.else8 ]
  54   ret i32 %retval.0
  55 }
  56
  57 ;      SM_52: .visible .func  (.param .b32 func_retval0) bar()
  58 ;      SM_52: mov.b32         %[[REG:.+]], 2;
  59 ; SM_52-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
  60 ; SM_52-NEXT: ret;
  61 ;
  62 ;      SM_70: .visible .func  (.param .b32 func_retval0) bar()
  63 ;      SM_70: mov.b32         %[[REG:.+]], 1;
  64 ; SM_70-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
  65 ; SM_70-NEXT: ret;
  66 ;
  67 ;      SM_90: .visible .func  (.param .b32 func_retval0) bar()
  68 ;      SM_90: mov.b32         %[[REG:.+]], 1;
  69 ; SM_90-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
  70 ; SM_90-NEXT: ret;
  71 define i32 @bar() {
  72 entry:
  73   %call = call i32 @__nvvm_reflect(ptr @.str)
  74   %cmp = icmp uge i32 %call, 700
  75   br i1 %cmp, label %if.then, label %if.else
  76
  77 if.then:
  78   br label %if.end
  79
  80 if.else:
  81   br label %if.end
  82
  83 if.end:
  84   %x = phi i32 [ 1, %if.then ], [ 2, %if.else ]
  85   ret i32 %x
  86 }
  87
  88 ; SM_52-NOT: valid;
  89 ; SM_70: valid;
  90 ; SM_90: valid;
  91 define void @baz() {
  92 entry:
  93   %call = call i32 @__nvvm_reflect(ptr @.str)
  94   %cmp = icmp uge i32 %call, 700
  95   br i1 %cmp, label %if.then, label %if.end
  96
  97 if.then:
  98   call void asm sideeffect "valid;\0A", ""()
  99   br label %if.end
 100
 101 if.end:
 102   ret void
 103 }
 104
 105 ;      SM_52: .visible .func  (.param .b32 func_retval0) qux()
 106 ;      SM_52: mov.b32         %[[REG:.+]], 3;
 107 ; SM_52-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
 108 ; SM_52-NEXT: ret;
 109 ;
 110 ;      SM_70: .visible .func  (.param .b32 func_retval0) qux()
 111 ;      SM_70: mov.b32         %[[REG:.+]], 2;
 112 ; SM_70-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
 113 ; SM_70-NEXT: ret;
 114 ;
 115 ;      SM_90: .visible .func  (.param .b32 func_retval0) qux()
 116 ;      SM_90: mov.b32         %[[REG:.+]], 1;
 117 ; SM_90-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
 118 ; SM_90-NEXT: ret;
 119 define i32 @qux() {
 120 entry:
 121   %call = call i32 @__nvvm_reflect(ptr noundef @.str)
 122   switch i32 %call, label %sw.default [
 123     i32 900, label %sw.bb
 124     i32 700, label %sw.bb1
 125     i32 520, label %sw.bb2
 126   ]
 127
 128 sw.bb:
 129   br label %return
 130
 131 sw.bb1:
 132   br label %return
 133
 134 sw.bb2:
 135   br label %return
 136
 137 sw.default:
 138   br label %return
 139
 140 return:
 141   %retval = phi i32 [ 4, %sw.default ], [ 3, %sw.bb2 ], [ 2, %sw.bb1 ], [ 1, %sw.bb ]
 142   ret i32 %retval
 143 }
 144
 145 ;      SM_52: .visible .func  (.param .b32 func_retval0) phi()
 146 ;      SM_52: mov.f32         %[[REG:.+]], 0f00000000;
 147 ; SM_52-NEXT: st.param.f32    [func_retval0+0], %[[REG]];
 148 ; SM_52-NEXT: ret;
 149 ;      SM_70: .visible .func  (.param .b32 func_retval0) phi()
 150 ;      SM_70: mov.f32         %[[REG:.+]], 0f00000000;
 151 ; SM_70-NEXT: st.param.f32    [func_retval0+0], %[[REG]];
 152 ; SM_70-NEXT: ret;
 153 ;      SM_90: .visible .func  (.param .b32 func_retval0) phi()
 154 ;      SM_90: mov.f32         %[[REG:.+]], 0f00000000;
 155 ; SM_90-NEXT: st.param.f32    [func_retval0+0], %[[REG]];
 156 ; SM_90-NEXT: ret;
 157 define float @phi() {
 158 entry:
 159   %0 = call i32 @__nvvm_reflect(ptr @.str)
 160   %1 = icmp eq i32 %0, 0
 161   br i1 %1, label %if.then, label %if.else
 162
 163 if.then:
 164   br label %if.else
 165
 166 if.else:
 167   %.08 = phi float [ 0.000000e+00, %if.then ], [ 1.000000e+00, %entry ]
 168   %4 = fcmp ogt float %.08, 0.000000e+00
 169   br i1 %4, label %exit, label %if.exit
 170
 171 if.exit:
 172   br label %exit
 173
 174 exit:
 175   ret float 0.000000e+00
 176 }
 177
 178 ;      SM_52: .visible .func  (.param .b32 func_retval0) prop()
 179 ;      SM_52: mov.b32         %[[REG:.+]], 3;
 180 ; SM_52-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
 181 ; SM_52-NEXT: ret;
 182 ;
 183 ;      SM_70: .visible .func  (.param .b32 func_retval0) prop()
 184 ;      SM_70: mov.b32         %[[REG:.+]], 2;
 185 ; SM_70-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
 186 ; SM_70-NEXT: ret;
 187 ;
 188 ;      SM_90: .visible .func  (.param .b32 func_retval0) prop()
 189 ;      SM_90: mov.b32         %[[REG:.+]], 1;
 190 ; SM_90-NEXT: st.param.b32    [func_retval0+0], %[[REG:.+]];
 191 ; SM_90-NEXT: ret;
 192 define i32 @prop() {
 193 entry:
 194   %call = call i32 @__nvvm_reflect(ptr @.str)
 195   %conv = zext i32 %call to i64
 196   %div = udiv i64 %conv, 100
 197   %cmp = icmp eq i64 %div, 9
 198   br i1 %cmp, label %if.then, label %if.else
 199
 200 if.then:
 201   br label %return
 202
 203 if.else:
 204   %div2 = udiv i64 %conv, 100
 205   %cmp3 = icmp eq i64 %div2, 7
 206   br i1 %cmp3, label %if.then5, label %if.else6
 207
 208 if.then5:
 209   br label %return
 210
 211 if.else6:
 212   %div7 = udiv i64 %conv, 100
 213   %cmp8 = icmp eq i64 %div7, 5
 214   br i1 %cmp8, label %if.then10, label %if.else11
 215
 216 if.then10:
 217   br label %return
 218
 219 if.else11:
 220   br label %return
 221
 222 return:
 223   %retval = phi i32 [ 1, %if.then ], [ 2, %if.then5 ], [ 3, %if.then10 ], [ 4, %if.else11 ]
 224   ret i32 %retval
 225 }