address comments

llvm · Aug 9, 2024 · b3fd15e · b3fd15e
1 parent e2d9a0a
commit b3fd15e
Showing 1 changed file with 101 additions and 0 deletions.
diff --git a/llvm/test/CodeGen/NVPTX/jump-table.ll b/llvm/test/CodeGen/NVPTX/jump-table.ll
@@ -67,3 +67,104 @@ case3:
 end:
   ret void
 }
+
+
+define i32 @test2(i32 %tmp158) {
+; CHECK-LABEL: test2(
+; CHECK:       {
+; CHECK-NEXT:    .reg .pred %p<6>;
+; CHECK-NEXT:    .reg .b32 %r<10>;
+; CHECK-EMPTY:
+; CHECK-NEXT:  // %bb.0: // %entry
+; CHECK-NEXT:    ld.param.u32 %r1, [test2_param_0];
+; CHECK-NEXT:    setp.gt.s32 %p1, %r1, 119;
+; CHECK-NEXT:    @%p1 bra $L__BB1_4;
+; CHECK-NEXT:  // %bb.1: // %entry
+; CHECK-NEXT:    setp.lt.u32 %p4, %r1, 6;
+; CHECK-NEXT:    @%p4 bra $L__BB1_3;
+; CHECK-NEXT:  // %bb.2: // %entry
+; CHECK-NEXT:    setp.lt.s32 %p5, %r1, -2147483645;
+; CHECK-NEXT:    @%p5 bra $L__BB1_3;
+; CHECK-NEXT:    bra.uni $L__BB1_6;
+; CHECK-NEXT:  $L__BB1_4: // %entry
+; CHECK-NEXT:    add.s32 %r2, %r1, -120;
+; CHECK-NEXT:    setp.gt.u32 %p2, %r2, 5;
+; CHECK-NEXT:    @%p2 bra $L__BB1_5;
+; CHECK-NEXT:  // %bb.12: // %entry
+; CHECK-NEXT:    $L_brx_0: .branchtargets
+; CHECK-NEXT:     $L__BB1_3,
+; CHECK-NEXT:     $L__BB1_7,
+; CHECK-NEXT:     $L__BB1_8,
+; CHECK-NEXT:     $L__BB1_9,
+; CHECK-NEXT:     $L__BB1_10,
+; CHECK-NEXT:     $L__BB1_11;
+; CHECK-NEXT:    brx.idx %r2, $L_brx_0;
+; CHECK-NEXT:  $L__BB1_7: // %bb339
+; CHECK-NEXT:    mov.b32 %r7, 12;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r7;
+; CHECK-NEXT:    ret;
+; CHECK-NEXT:  $L__BB1_5: // %entry
+; CHECK-NEXT:    setp.eq.s32 %p3, %r1, 1024;
+; CHECK-NEXT:    @%p3 bra $L__BB1_3;
+; CHECK-NEXT:    bra.uni $L__BB1_6;
+; CHECK-NEXT:  $L__BB1_3: // %bb338
+; CHECK-NEXT:    mov.b32 %r8, 11;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r8;
+; CHECK-NEXT:    ret;
+; CHECK-NEXT:  $L__BB1_10: // %bb342
+; CHECK-NEXT:    mov.b32 %r4, 15;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r4;
+; CHECK-NEXT:    ret;
+; CHECK-NEXT:  $L__BB1_6: // %bb336
+; CHECK-NEXT:    mov.b32 %r9, 10;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r9;
+; CHECK-NEXT:    ret;
+; CHECK-NEXT:  $L__BB1_8: // %bb340
+; CHECK-NEXT:    mov.b32 %r6, 13;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r6;
+; CHECK-NEXT:    ret;
+; CHECK-NEXT:  $L__BB1_9: // %bb341
+; CHECK-NEXT:    mov.b32 %r5, 14;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r5;
+; CHECK-NEXT:    ret;
+; CHECK-NEXT:  $L__BB1_11: // %bb343
+; CHECK-NEXT:    mov.b32 %r3, 18;
+; CHECK-NEXT:    st.param.b32 [func_retval0+0], %r3;
+; CHECK-NEXT:    ret;
+entry:
+  switch i32 %tmp158, label %bb336 [
+    i32 -2147483648, label %bb338
+    i32 -2147483647, label %bb338
+    i32 -2147483646, label %bb338
+    i32 120, label %bb338
+    i32 121, label %bb339
+    i32 122, label %bb340
+    i32 123, label %bb341
+    i32 124, label %bb342
+    i32 125, label %bb343
+    i32 126, label %bb336
+    i32 1024, label %bb338
+    i32 0, label %bb338
+    i32 1, label %bb338
+    i32 2, label %bb338
+    i32 3, label %bb338
+    i32 4, label %bb338
+    i32 5, label %bb338
+  ]
+
+bb336:
+  ret i32 10
+bb338:
+  ret i32 11
+bb339:
+  ret i32 12
+bb340:
+  ret i32 13
+bb341:
+  ret i32 14
+bb342:
+  ret i32 15
+bb343:
+  ret i32 18
+
+}