diff --git a/llvm/lib/Analysis/LoopAccessAnalysis.cpp b/llvm/lib/Analysis/LoopAccessAnalysis.cpp --- a/llvm/lib/Analysis/LoopAccessAnalysis.cpp +++ b/llvm/lib/Analysis/LoopAccessAnalysis.cpp @@ -1042,6 +1042,12 @@ return 0; } + if (isa(AccessTy)) { + LLVM_DEBUG(dbgs() << "LAA: Unsupported stride - scalable object: " + << *AccessTy << "\n"); + return 0; + } + const SCEV *PtrScev = replaceSymbolicStrideSCEV(PSE, StridesMap, Ptr); const SCEVAddRecExpr *AR = dyn_cast(PtrScev); @@ -1101,7 +1107,8 @@ } auto &DL = Lp->getHeader()->getModule()->getDataLayout(); - int64_t Size = DL.getTypeAllocSize(AccessTy); + TypeSize AllocSize = DL.getTypeAllocSize(AccessTy); + int64_t Size = AllocSize.getFixedSize(); const APInt &APStepVal = C->getAPInt(); // Huge step value - give up. diff --git a/llvm/test/Analysis/LoopAccessAnalysis/gep-induction-operand-typesize-warning.ll b/llvm/test/Analysis/LoopAccessAnalysis/gep-induction-operand-typesize-warning.ll deleted file mode 100644 --- a/llvm/test/Analysis/LoopAccessAnalysis/gep-induction-operand-typesize-warning.ll +++ /dev/null @@ -1,21 +0,0 @@ -; RUN: opt -loop-load-elim -mtriple=aarch64--linux-gnu -mattr=+sve < %s - -; This regression test is verifying that a GEP instruction performed on a -; scalable vector does not produce a 'assumption that TypeSize is not scalable' -; warning in the llvm::getGEPInductionOperand function. - -define void @get_gep_induction_operand_typesize_warning(i64 %n, * %a) { -entry: - br label %loop.body - -loop.body: - %0 = phi i64 [ 0, %entry ], [ %1, %loop.body ] - %idx = getelementptr , * %a, i64 %0 - store zeroinitializer, * %idx - %1 = add i64 %0, 1 - %2 = icmp eq i64 %1, %n - br i1 %2, label %loop.end, label %loop.body - -loop.end: - ret void -} diff --git a/llvm/test/Analysis/LoopAccessAnalysis/scalable-vector-regression-tests.ll b/llvm/test/Analysis/LoopAccessAnalysis/scalable-vector-regression-tests.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Analysis/LoopAccessAnalysis/scalable-vector-regression-tests.ll @@ -0,0 +1,54 @@ +; RUN: opt -loop-load-elim -mtriple=aarch64 -mattr=+sve -S < %s + +; Regression tests verifying "assumption that TypeSize is not scalable" and +; "Invalid size request on a scalable vector." are not produced by +; -load-loop-elim. + +define void @regression_test_get_gep_induction_operand_typesize_warning(i64 %n, * %a) { +entry: + br label %loop.body + +loop.body: + %0 = phi i64 [ 0, %entry ], [ %1, %loop.body ] + %idx = getelementptr , * %a, i64 %0 + store zeroinitializer, * %idx + %1 = add i64 %0, 1 + %2 = icmp eq i64 %1, %n + br i1 %2, label %loop.end, label %loop.body + +loop.end: + ret void +} + +define void @regression_test_loop_access_scalable_typesize(* %input_ptr) { +entry: + br label %vector.body +vector.body: + %ind_ptr = phi * [ %next_ptr, %vector.body ], [ %input_ptr, %entry ] + %ind = phi i64 [ %next, %vector.body ], [ 0, %entry ] + %ld = load , * %ind_ptr, align 16 + store zeroinitializer, * %ind_ptr, align 16 + %next_ptr = getelementptr inbounds , * %ind_ptr, i64 1 + %next = add i64 %ind, 1 + %cond = icmp ult i64 %next, 1024 + br i1 %cond, label %end, label %vector.body +end: + ret void +} + +define void @regression_test_loop_access_scalable_typesize_nonscalable_object(i8* %input_ptr) { +entry: + br label %vector.body +vector.body: + %ind_ptr = phi i8* [ %next_ptr, %vector.body ], [ %input_ptr, %entry ] + %ind = phi i64 [ %next, %vector.body ], [ 0, %entry ] + %scalable_ptr = bitcast i8* %ind_ptr to * + %ld = load , * %scalable_ptr, align 16 + store zeroinitializer, * %scalable_ptr, align 16 + %next_ptr = getelementptr inbounds i8, i8* %ind_ptr, i64 1 + %next = add i64 %ind, 1 + %cond = icmp ult i64 %next, 1024 + br i1 %cond, label %end, label %vector.body +end: + ret void +}