Fix the bug when handling shufflevector for aarch64.

[danielcdh]

Summary

This Fixes https://bugs.llvm.org/show_bug.cgi?id=33600

[mssimpso]

Hi Dehao,

Thanks for fixing this. Can you move the test to Transforms/InterleavedAccess/AArch64/interleaved-accesses.ll and add some appropriate CHECK lines to make sure we're generating the correct code?

[danielcdh]

move the test to Transforms/InterleavedAccess/AArch64/interleaved-accesses.ll

[mssimpso]

LGTM.

[chandlerc]

test/CodeGen/AArch64/pr33600.ll

1

Please actually run the output through FileCheck. This kind of test in particular you can almost certainly generate nice checks with the utils/update_llc_test_checks.py script.

1

Ignore this, I should learn to refresh my browser. Sorry for the noise.

lib/Target/AArch64/AArch64ISelLowering.cpp

for (unsigned LoadCount = 0; LoadCount < NumLoads; ++LoadCount) {
     for (unsigned i = 0; i < Shuffles.size(); i++) {
       ShuffleVectorInst *SVI = Shuffles[i];
       unsigned Index = Indices[i];
 
       Value *SubVec = Builder.CreateExtractValue(LdN, Index);
 
       // Convert the integer vector to pointer vector if the element is pointer.
       if (EltTy->isPointerTy())
-        SubVec = Builder.CreateIntToPtr(SubVec, SVI->getType());
-
+        SubVec = Builder.CreateIntToPtr(
+            SubVec, VectorType::get(SVI->getType()->getVectorElementType(),
+                                    VecTy->getVectorNumElements()));
       SubVecs[SVI].push_back(SubVec);
     }
   }
 
   // Replace uses of the shufflevector instructions with the sub-vectors
   // returned by the load intrinsic. If a shufflevector instruction is
   // associated with more than one sub-vector, those sub-vectors will be
   // concatenated into a single wide vector.

test/CodeGen/AArch64/pr33600.ll

+; RUN: llc < %s
+
+target triple = "aarch64-unknown-linux-gnu"
+
+define <4 x i1> @foo(<12 x i64 *>* %p) {
+  %l = load <12 x i64 *>, <12 x i64 *>* %p
+  %s1 = shufflevector <12 x i64 *> %l, <12 x i64 *> undef, <4 x i32> <i32 0, i32 3, i32 6, i32 9>
+  %s2 = shufflevector <12 x i64 *> %l, <12 x i64 *> undef, <4 x i32> <i32 1, i32 4, i32 7, i32 10>
+  %ret = icmp ne <4 x i64 *> %s1, %s2
+  ret <4 x i1> %ret
+}