Simplify LShr construction; add AArch64/PowerPC tests

Author: uyoyo0

llvm/lib/Transforms/Vectorize/VectorCombine.cpp

@@ -2021,8 +2021,7 @@ bool VectorCombine::scalarizeExtExtract(Instruction &I) {
     uint64_t ShiftAmt = DL->isBigEndian()
         ? (TotalBits - SrcEltSizeInBits - Idx * SrcEltSizeInBits)
         : (Idx * SrcEltSizeInBits);
-    Value *ShAmtVal = ConstantInt::get(PackedTy, ShiftAmt);
-    Value *LShr = Builder.CreateLShr(ScalarV, ShAmtVal);
+    Value *LShr = Builder.CreateLShr(ScalarV, ShiftAmt);
     Value *And = Builder.CreateAnd(LShr, Mask);
     U->replaceAllUsesWith(And);
   }

llvm/test/Transforms/VectorCombine/scalarize-ext-extract-endian.ll renamed to llvm/test/Transforms/VectorCombine/AArch64/scalarize-ext-extract-endian.ll

@@ -1,5 +1,5 @@
 ; RUN: opt -passes='vector-combine,dce' -S -mtriple=aarch64-unknown-linux-gnu %s -o - | FileCheck %s --check-prefix=LE
-; RUN: opt -passes='vector-combine,dce' -S -mtriple=powerpc64-ibm-aix-xcoff   %s -o - | FileCheck %s --check-prefix=BE
+; RUN: opt -passes='vector-combine,dce' -S -mtriple=aarch64_be-unknown-linux-gnu %s -o - | FileCheck %s --check-prefix=BE
 
 define i64 @g(<8 x i8> %v) {
   %z  = zext <8 x i8> %v to <8 x i64>

llvm/test/Transforms/VectorCombine/PowerPC/scalarize-ext-extract.ll

@@ -0,0 +1,23 @@
+; RUN: opt -passes='vector-combine,dce' -S -mtriple=aarch64-unknown-linux-gnu %s -o - | FileCheck %s --check-prefix=LE
+; RUN: opt -passes='vector-combine,dce' -S -mtriple=powerpc64-ibm-aix-xcoff     %s -o - | FileCheck %s --check-prefix=BE
+
+define i64 @g(<8 x i8> %v) {
+  %z  = zext <8 x i8> %v to <8 x i64>
+  %e0 = extractelement <8 x i64> %z, i32 0
+  %e7 = extractelement <8 x i64> %z, i32 7
+  %sum = add i64 %e0, %e7
+  ret i64 %sum
+}
+
+; LE-LABEL: @g(
+; LE: bitcast <8 x i8> %{{.*}} to i64
+; LE: lshr i64 %{{.*}}, 56
+; LE: and i64 %{{.*}}, 255
+; LE-NOT: extractelement
+
+; BE-LABEL: @g(
+; BE: bitcast <8 x i8> %{{.*}} to i64
+; BE: and i64 %{{.*}}, 255
+; BE: lshr i64 %{{.*}}, 56
+; BE-NOT: extractelement
+