diff --git a/llvm/lib/Transforms/Scalar/InstructionCombining.cpp b/llvm/lib/Transforms/Scalar/InstructionCombining.cpp index 21b51e4cda12..94547b48bb5b 100644 --- a/llvm/lib/Transforms/Scalar/InstructionCombining.cpp +++ b/llvm/lib/Transforms/Scalar/InstructionCombining.cpp @@ -9033,6 +9033,29 @@ Instruction *InstCombiner::visitBitCast(BitCastInst &CI) { } } + if (const VectorType *DestVTy = dyn_cast(DestTy)) { + if (DestVTy->getNumElements() == 1) { + if (!isa(SrcTy)) { + Value *Elem = InsertCastBefore(Instruction::BitCast, Src, + DestVTy->getElementType(), CI); + return InsertElementInst::Create(Context->getUndef(DestTy), Elem, + Context->getNullValue(Type::Int32Ty)); + } + // FIXME: Canonicalize bitcast(insertelement) -> insertelement(bitcast) + } + } + + if (const VectorType *SrcVTy = dyn_cast(SrcTy)) { + if (SrcVTy->getNumElements() == 1) { + if (!isa(DestTy)) { + Instruction *Elem = + new ExtractElementInst(Src, Context->getNullValue(Type::Int32Ty)); + InsertNewInstBefore(Elem, CI); + return CastInst::Create(Instruction::BitCast, Elem, DestTy); + } + } + } + if (ShuffleVectorInst *SVI = dyn_cast(Src)) { if (SVI->hasOneUse()) { // Okay, we have (bitconvert (shuffle ..)). Check to see if this is @@ -12477,6 +12500,7 @@ Instruction *InstCombiner::visitExtractElementInst(ExtractElementInst &EI) { Context->getConstantInt(Type::Int32Ty, SrcIdx, false)); } } + // FIXME: Canonicalize extractelement(bitcast) -> bitcast(extractelement) } return 0; } diff --git a/llvm/test/Transforms/InstCombine/bitcast-vec-canon.ll b/llvm/test/Transforms/InstCombine/bitcast-vec-canon.ll new file mode 100644 index 000000000000..6bb986a2eb94 --- /dev/null +++ b/llvm/test/Transforms/InstCombine/bitcast-vec-canon.ll @@ -0,0 +1,22 @@ +; RUN: llvm-as < %s | opt -instcombine | llvm-dis | grep element | count 4 + +define double @a(<1 x i64> %y) { + %c = bitcast <1 x i64> %y to double + ret double %c +} + +define i64 @b(<1 x i64> %y) { + %c = bitcast <1 x i64> %y to i64 + ret i64 %c +} + +define <1 x i64> @c(double %y) { + %c = bitcast double %y to <1 x i64> + ret <1 x i64> %c +} + +define <1 x i64> @d(i64 %y) { + %c = bitcast i64 %y to <1 x i64> + ret <1 x i64> %c +} +