[llvm] 5a64bc2 - [DAGCombiner] Remove overzealous assertion when folding assert+trunc+assert (PR55846)

[Nikita Popov via llvm-commits]

Author: Nikita Popov
Date: 2022-06-07T09:50:26+02:00
New Revision: 5a64bc207ee07d939271ee8631bf28b01e431a46

URL: https://github.com/llvm/llvm-project/commit/5a64bc207ee07d939271ee8631bf28b01e431a46
DIFF: https://github.com/llvm/llvm-project/commit/5a64bc207ee07d939271ee8631bf28b01e431a46.diff

LOG: [DAGCombiner] Remove overzealous assertion when folding assert+trunc+assert (PR55846)

These assert that there are no "useless" assertzext/assertsext nodes
(that assert a wider width than a following trunc), but I don't think
there is anything preventing such nodes from reaching this code.
I don't think the assertion is relevant for correctness of this
transform either -- if such an assert is present, then the other
one will always be to a smaller width, and we'll pick that one.
The assertion dates back to D37017.

Fixes https://github.com/llvm/llvm-project/issues/55846.

Differential Revision: https://reviews.llvm.org/D126952

Added: 
    llvm/test/CodeGen/X86/pr55846.ll

Modified: 
    llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp b/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
index eb14a8ab1ecdf..ebde92a70f786 100644
--- a/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
@@ -12508,13 +12508,9 @@ SDValue DAGCombiner::visitAssertExt(SDNode *N) {
     // This eliminates the later assert:
     // assert (trunc (assert X, i8) to iN), i1 --> trunc (assert X, i1) to iN
     // assert (trunc (assert X, i1) to iN), i8 --> trunc (assert X, i1) to iN
+    SDLoc DL(N);
     SDValue BigA = N0.getOperand(0);
     EVT BigA_AssertVT = cast<VTSDNode>(BigA.getOperand(1))->getVT();
-    assert(BigA_AssertVT.bitsLE(N0.getValueType()) &&
-           "Asserting zero/sign-extended bits to a type larger than the "
-           "truncated destination does not provide information");
-
-    SDLoc DL(N);
     EVT MinAssertVT = AssertVT.bitsLT(BigA_AssertVT) ? AssertVT : BigA_AssertVT;
     SDValue MinAssertVTVal = DAG.getValueType(MinAssertVT);
     SDValue NewAssert = DAG.getNode(Opcode, DL, BigA.getValueType(),
@@ -12530,10 +12526,6 @@ SDValue DAGCombiner::visitAssertExt(SDNode *N) {
       Opcode == ISD::AssertZext) {
     SDValue BigA = N0.getOperand(0);
     EVT BigA_AssertVT = cast<VTSDNode>(BigA.getOperand(1))->getVT();
-    assert(BigA_AssertVT.bitsLE(N0.getValueType()) &&
-           "Asserting zero/sign-extended bits to a type larger than the "
-           "truncated destination does not provide information");
-
     if (AssertVT.bitsLT(BigA_AssertVT)) {
       SDLoc DL(N);
       SDValue NewAssert = DAG.getNode(Opcode, DL, BigA.getValueType(),

diff  --git a/llvm/test/CodeGen/X86/pr55846.ll b/llvm/test/CodeGen/X86/pr55846.ll
new file mode 100644
index 0000000000000..4267ef4b0939d
--- /dev/null
+++ b/llvm/test/CodeGen/X86/pr55846.ll
@@ -0,0 +1,26 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
+; RUN: llc -mtriple=x86_64-unknown-linux-gnu < %s | FileCheck %s
+
+; After legalization, this could be: "i8 truncate (i64 AssertZext X, Type: i9)"
+; The AssertZext does not add information, so it should be eliminated,
+; but that must not trigger a compile-time assert.
+
+define void @test(i64* %p) {
+; CHECK-LABEL: test:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    movl $256, %eax # imm = 0x100
+; CHECK-NEXT:    movq %rax, (%rdi)
+; CHECK-NEXT:    retq
+  %sel = select i1 true, i64 256, i64 0
+  br label %bb2
+
+bb2:
+  store i64 %sel, i64* %p, align 4
+  %p.bc = bitcast i64* %p to <2 x i1>*
+  %load = load <2 x i1>, <2 x i1>* %p.bc, align 1
+  br label %bb3
+
+bb3:
+  %use = add <2 x i1> %load, zeroinitializer
+  ret void
+}