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 @@ -1057,6 +1057,12 @@ assert(Ty->isPointerTy() && "Unexpected non-ptr"); assert(!AccessTy->isAggregateType() && "Bad stride - Not a pointer to a scalar type"); + if (isa(AccessTy)) { + LLVM_DEBUG(dbgs() << "LAA: Bad stride - Scalable object: " << *AccessTy + << "\n"); + return 0; + } + const SCEV *PtrScev = replaceSymbolicStrideSCEV(PSE, StridesMap, Ptr); const SCEVAddRecExpr *AR = dyn_cast(PtrScev); @@ -1117,7 +1123,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,62 @@ +; RUN: opt -loop-load-elim -mtriple=aarch64 -mattr=+sve -S -debug < %s 2>&1 | FileCheck %s +; REQUIRES: asserts + +; Regression tests verifying "assumption that TypeSize is not scalable" and +; "Invalid size request on a scalable vector." are not produced by +; -load-loop-elim (this would cause the test to fail because opt would exit with +; a non-zero exit status). + +; No output checked for this one, but causes a fatal error if the regression is present. + +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 +} + +; CHECK-LABEL: LAA: Found a loop in regression_test_loop_access_scalable_typesize +; CHECK: LAA: Bad stride - Scalable object: +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 +} + +; CHECK-LABEL: LAA: Found a loop in regression_test_loop_access_scalable_typesize_nonscalable_object +; CHECK: LAA: Bad stride - Scalable object: +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 +}