From d7b6ea166dfcf6fc39e953a9b81bf84d6927053e Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Fri, 5 Jan 2007 07:36:08 +0000 Subject: [PATCH] Implement InstCombine/vec_shuffle.ll:%test7, simplifying shuffles with undef operands. llvm-svn: 32899 --- .../Scalar/InstructionCombining.cpp | 24 ++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/llvm/lib/Transforms/Scalar/InstructionCombining.cpp b/llvm/lib/Transforms/Scalar/InstructionCombining.cpp index c1a331faecb6..eaab4c6b9c58 100644 --- a/llvm/lib/Transforms/Scalar/InstructionCombining.cpp +++ b/llvm/lib/Transforms/Scalar/InstructionCombining.cpp @@ -8833,8 +8833,30 @@ Instruction *InstCombiner::visitShuffleVectorInst(ShuffleVectorInst &SVI) { if (isa(SVI.getOperand(2))) return ReplaceInstUsesWith(SVI, UndefValue::get(SVI.getType())); - // TODO: If we have shuffle(x, undef, mask) and any elements of mask refer to + // If we have shuffle(x, undef, mask) and any elements of mask refer to // the undef, change them to undefs. + if (isa(SVI.getOperand(1))) { + // Scan to see if there are any references to the RHS. If so, replace them + // with undef element refs and set MadeChange to true. + for (unsigned i = 0, e = Mask.size(); i != e; ++i) { + if (Mask[i] >= e && Mask[i] != 2*e) { + Mask[i] = 2*e; + MadeChange = true; + } + } + + if (MadeChange) { + // Remap any references to RHS to use LHS. + std::vector Elts; + for (unsigned i = 0, e = Mask.size(); i != e; ++i) { + if (Mask[i] == 2*e) + Elts.push_back(UndefValue::get(Type::Int32Ty)); + else + Elts.push_back(ConstantInt::get(Type::Int32Ty, Mask[i])); + } + SVI.setOperand(2, ConstantPacked::get(Elts)); + } + } // Canonicalize shuffle(x ,x,mask) -> shuffle(x, undef,mask') // Canonicalize shuffle(undef,x,mask) -> shuffle(x, undef,mask'). -- GitLab