https://github.com/llvmbot created https://github.com/llvm/llvm-project/pull/177138
Backport 242ca4e116d18849187617d7399be20b136d768b Requested by: @davemgreen >From a536d8cec1f3b5468796c624be497bcdf9b16404 Mon Sep 17 00:00:00 2001 From: David Green <[email protected]> Date: Wed, 21 Jan 2026 09:14:34 +0000 Subject: [PATCH] [AArch64] Protect against unexpected SIGN_EXTEND_INREG in performBuildShuffleExtendCombine (#176733) Apparently this code is only expecting shuffle of SIGN_EXTEND or ZERO_EXTEND, but can sometimes see a SIGN_EXTEND_INREG of the second vector operand. Add a check that the second operand has the same constraints as the first. Fixes #176314 (cherry picked from commit 242ca4e116d18849187617d7399be20b136d768b) --- .../Target/AArch64/AArch64ISelLowering.cpp | 4 ++++ .../CodeGen/AArch64/aarch64-dup-ext-crash.ll | 20 +++++++++++++++++++ 2 files changed, 24 insertions(+) diff --git a/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp b/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp index 74ee8ff8ab5f5..cd22caabe259a 100644 --- a/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp +++ b/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp @@ -19834,6 +19834,10 @@ static SDValue performBuildShuffleExtendCombine(SDValue BV, SelectionDAG &DAG) { return SDValue(); unsigned Opc = Op.getOpcode(); + if (BV.getOpcode() == ISD::VECTOR_SHUFFLE && + (Opc != ISD::SIGN_EXTEND && Opc != ISD::ZERO_EXTEND)) + return SDValue(); + if (Opc == ISD::ANY_EXTEND) continue; diff --git a/llvm/test/CodeGen/AArch64/aarch64-dup-ext-crash.ll b/llvm/test/CodeGen/AArch64/aarch64-dup-ext-crash.ll index 478c1be8821f6..eb75ddf9250a6 100644 --- a/llvm/test/CodeGen/AArch64/aarch64-dup-ext-crash.ll +++ b/llvm/test/CodeGen/AArch64/aarch64-dup-ext-crash.ll @@ -86,3 +86,23 @@ vector.body: ; preds = %vector.body, %vecto end: ret i32 %and255 } + +define <4 x i32> @backsmith_pure_4(<4 x i32> %0, <4 x i16> %conv.i) { +; CHECK-LABEL: backsmith_pure_4: +; CHECK: // %bb.0: // %entry +; CHECK-NEXT: // kill: def $d1 killed $d1 def $q1 +; CHECK-NEXT: shl v2.4s, v0.4s, #16 +; CHECK-NEXT: dup v1.4h, v1.h[2] +; CHECK-NEXT: sshr v2.4s, v2.4s, #16 +; CHECK-NEXT: sshll v1.4s, v1.4h, #0 +; CHECK-NEXT: trn2 v1.4s, v1.4s, v2.4s +; CHECK-NEXT: mul v0.4s, v0.4s, v1.4s +; CHECK-NEXT: ret +entry: + %conv.i19 = trunc <4 x i32> %0 to <4 x i16> + %shuffle11 = shufflevector <4 x i16> %conv.i19, <4 x i16> %conv.i, <32 x i32> <i32 5, i32 2, i32 5, i32 3, i32 1, i32 6, i32 5, i32 6, i32 5, i32 6, i32 4, i32 6, i32 0, i32 3, i32 5, i32 1, i32 6, i32 7, i32 2, i32 6, i32 3, i32 1, i32 0, i32 4, i32 4, i32 1, i32 2, i32 7, i32 2, i32 2, i32 1, i32 2> + %conv12 = sext <32 x i16> %shuffle11 to <32 x i32> + %shuffle14 = shufflevector <32 x i32> %conv12, <32 x i32> zeroinitializer, <4 x i32> <i32 9, i32 25, i32 poison, i32 poison> + %mul = mul <4 x i32> %0, %shuffle14 + ret <4 x i32> %mul +} _______________________________________________ llvm-branch-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits
