Index: llvm/trunk/lib/IR/Instructions.cpp =================================================================== --- llvm/trunk/lib/IR/Instructions.cpp +++ llvm/trunk/lib/IR/Instructions.cpp @@ -2516,17 +2516,19 @@ { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,13,12}, // AddrSpaceCast -+ }; + // TODO: This logic could be encoded into the table above and handled in the + // switch below. // If either of the casts are a bitcast from scalar to vector, disallow the - // merging. However, bitcast of A->B->A are allowed. - bool isFirstBitcast = (firstOp == Instruction::BitCast); - bool isSecondBitcast = (secondOp == Instruction::BitCast); - bool chainedBitcast = (SrcTy == DstTy && isFirstBitcast && isSecondBitcast); - - // Check if any of the bitcasts convert scalars<->vectors. - if ((isFirstBitcast && isa(SrcTy) != isa(MidTy)) || - (isSecondBitcast && isa(MidTy) != isa(DstTy))) - // Unless we are bitcasting to the original type, disallow optimizations. - if (!chainedBitcast) return 0; + // merging. However, any pair of bitcasts are allowed. + bool IsFirstBitcast = (firstOp == Instruction::BitCast); + bool IsSecondBitcast = (secondOp == Instruction::BitCast); + bool AreBothBitcasts = IsFirstBitcast && IsSecondBitcast; + + // Check if any of the casts convert scalars <-> vectors. + if ((IsFirstBitcast && isa(SrcTy) != isa(MidTy)) || + (IsSecondBitcast && isa(MidTy) != isa(DstTy))) + if (!AreBothBitcasts) + return 0; int ElimCase = CastResults[firstOp-Instruction::CastOpsBegin] [secondOp-Instruction::CastOpsBegin]; Index: llvm/trunk/test/Transforms/InstCombine/bitcast-bitcast.ll =================================================================== --- llvm/trunk/test/Transforms/InstCombine/bitcast-bitcast.ll +++ llvm/trunk/test/Transforms/InstCombine/bitcast-bitcast.ll @@ -18,8 +18,7 @@ ret <2 x i32> %bc2 ; CHECK-LABEL: @bitcast_bitcast_s_s_v( -; CHECK-NEXT: %bc1 = bitcast i64 %a to double -; CHECK-NEXT: %bc2 = bitcast double %bc1 to <2 x i32> +; CHECK-NEXT: %bc2 = bitcast i64 %a to <2 x i32> ; CHECK-NEXT: ret <2 x i32> %bc2 } @@ -29,8 +28,7 @@ ret double %bc2 ; CHECK-LABEL: @bitcast_bitcast_s_v_s( -; CHECK-NEXT: %bc1 = bitcast i64 %a to <2 x i32> -; CHECK-NEXT: %bc2 = bitcast <2 x i32> %bc1 to double +; CHECK-NEXT: %bc2 = bitcast i64 %a to double ; CHECK-NEXT: ret double %bc2 } @@ -40,8 +38,7 @@ ret <2 x i32> %bc2 ; CHECK-LABEL: @bitcast_bitcast_s_v_v( -; CHECK-NEXT: %bc1 = bitcast i64 %a to <4 x i16> -; CHECK-NEXT: %bc2 = bitcast <4 x i16> %bc1 to <2 x i32> +; CHECK-NEXT: %bc2 = bitcast i64 %a to <2 x i32> ; CHECK-NEXT: ret <2 x i32> %bc2 } @@ -51,8 +48,7 @@ ret i64 %bc2 ; CHECK-LABEL: @bitcast_bitcast_v_s_s( -; CHECK-NEXT: %bc1 = bitcast <2 x i32> %a to double -; CHECK-NEXT: %bc2 = bitcast double %bc1 to i64 +; CHECK-NEXT: %bc2 = bitcast <2 x i32> %a to i64 ; CHECK-NEXT: ret i64 %bc2 } @@ -62,8 +58,7 @@ ret <4 x i16> %bc2 ; CHECK-LABEL: @bitcast_bitcast_v_s_v( -; CHECK-NEXT: %bc1 = bitcast <2 x i32> %a to double -; CHECK-NEXT: %bc2 = bitcast double %bc1 to <4 x i16> +; CHECK-NEXT: %bc2 = bitcast <2 x i32> %a to <4 x i16> ; CHECK-NEXT: ret <4 x i16> %bc2 } @@ -73,8 +68,7 @@ ret double %bc2 ; CHECK-LABEL: @bitcast_bitcast_v_v_s( -; CHECK-NEXT: %bc1 = bitcast <2 x float> %a to <4 x i16> -; CHECK-NEXT: %bc2 = bitcast <4 x i16> %bc1 to double +; CHECK-NEXT: %bc2 = bitcast <2 x float> %a to double ; CHECK-NEXT: ret double %bc2 }