llvm-for-llvmta/test/Transforms/LoopVectorize/AArch64/sve-scalable-load-in-loop.ll

50 lines
1.5 KiB
LLVM

; RUN: opt -S -loop-vectorize -mtriple=aarch64-linux-gnu -mattr=+sve < %s 2>%t | FileCheck %s
; RUN: FileCheck --check-prefix=WARN --allow-empty %s <%t
; This test is checking that a scalable load inside a loop does not trigger a
; TypeSize error in the loop vectorization legality analysis. It is possible for
; a scalable/vector load to appear inside a loop at vectorization legality
; analysis if, for example, the ACLE are used. If we encounter a scalable/vector
; load, it should not be considered for analysis, and we should not see a
; TypeSize error.
; If this check fails please read test/CodeGen/AArch64/README for instructions on how to resolve it.
; WARN-NOT: warning:
; #include <arm_sve.h>
;
; void scalable_load_in_loop(long n, int *a, int *b, svuint32_t *x,
; svuint32_t *y) {
; for (unsigned i = 0; i < n; i++) {
; if (i % 2 == 0) continue;
; a[i] = 2 * b[i];
; *x = *y;
; }
; }
; CHECK-LABEL: @scalable_load_in_loop
; CHECK-NOT: vector.body
define void @scalable_load_in_loop(i64 %n, <vscale x 4 x i32>* %x, <vscale x 4 x i32>* %y) {
entry:
br label %for.body
for.body:
%i = phi i32 [ %inc, %for.inc ], [ 0, %entry ]
%rem = and i32 %i, 1
%cmp = icmp eq i32 %rem, 0
br i1 %cmp, label %for.inc, label %if.end
if.end:
%0 = load <vscale x 4 x i32>, <vscale x 4 x i32>* %y
store <vscale x 4 x i32> %0, <vscale x 4 x i32>* %x
br label %for.inc
for.inc:
%inc = add i32 %i, 1
%cmp2 = icmp slt i64 0, %n
br i1 %cmp2, label %for.body, label %for.cleanup
for.cleanup:
ret void
}