Plus zext-zext case

Signed-off-by: Alexander Peskov <[email protected]>

Author: apeskov

llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp

@@ -6266,7 +6266,9 @@ SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
     if (OpOpcode == ISD::ZERO_EXTEND) { // (zext (zext x)) -> (zext x)
       SDNodeFlags Flags;
       Flags.setNonNeg(N1->getFlags().hasNonNeg());
-      return getNode(ISD::ZERO_EXTEND, DL, VT, N1.getOperand(0), Flags);
+      SDValue NewVal = getNode(ISD::ZERO_EXTEND, DL, VT, N1.getOperand(0), Flags);
+      transferDbgValues(N1, NewVal);
+      return NewVal;
     }
     if (N1.isUndef())
       // zext(undef) = 0, because the top bits will be zero.

llvm/test/DebugInfo/NVPTX/debug-bool-var.ll

@@ -5,9 +5,16 @@ declare void @foo(i32)
 define void @test1(i32 noundef %gid) !dbg !3 {
 entry:
   ;
-  ; verify that debug info exists for "xyz" variable
+  ; Equivalent of code:
+  ;   extern void foo(int);
+  ;   void test_kernel_bool(int a) {
+  ;     bool xyz = a == 0;
+  ;     foo(xyz);
+  ;   }
   ;
-  ; CHECK-LABEL:    DW_TAG_variable
+  ; Verify that debug info exists for "xyz" variable
+  ;
+  ; CHECK:       DW_TAG_variable
   ; CHECK:      .b8 120     // DW_AT_name
   ; CHECK-NEXT: .b8 121
   ; CHECK-NEXT: .b8 122
@@ -24,6 +31,35 @@ entry:
   ret void
 }
 
+define void @test2(i32 noundef %gid) !dbg !14 {
+entry:
+  ;
+  ; Equivalent of code:
+  ;   extern void foo(int);
+  ;   void test_kernel_bool(int a) {
+  ;     unsigned char abc = a == 0;
+  ;     foo(abc);
+  ;   }
+  ;
+  ; Verify that debug info exists for "abc" variable
+  ;
+  ; CHECK:       DW_TAG_variable
+  ; CHECK:      .b8 97      // DW_AT_name
+  ; CHECK-NEXT: .b8 98
+  ; CHECK-NEXT: .b8 99
+  ; CHECK-NEXT: .b8 0
+  ; CHECK-NEXT: .b8 1       // DW_AT_decl_file
+  ; CHECK-NEXT: .b8 11       // DW_AT_decl_line
+  ;
+  %cmp = icmp eq i32 %gid, 0, !dbg !17
+  %conv = zext i1 %cmp to i32, !dbg !17
+  %conv1 = trunc i32 %conv to i8, !dbg !17
+    #dbg_value(i8 %conv1, !16, !DIExpression(), !18)
+  %conv3 = zext i8 %conv1 to i32
+  call void @foo(i32 %conv3)
+  ret void
+}
+
 !llvm.dbg.cu = !{!0}
 !llvm.module.flags = !{!2}
 
@@ -41,3 +77,8 @@ entry:
 !11 = !DIBasicType(name: "bool", size: 8, encoding: DW_ATE_boolean)
 !12 = !DILocation(line: 1, column: 3, scope: !9)
 !13 = !DILocation(line: 2, scope: !9)
+!14 = distinct !DISubprogram(name: "test2", linkageName: "_test2i", scope: !1, file: !1, line: 10, type: !4, scopeLine: 10, unit: !0, retainedNodes: !8)
+!15 = distinct !DILexicalBlock(scope: !14, file: !1, line: 10, column: 30)
+!16 = !DILocalVariable(name: "abc", scope: !15, file: !1, line: 11, type: !11)
+!17 = !DILocation(line: 11, column: 3, scope: !15)
+!18 = !DILocation(line: 12, scope: !15)