xref: /llvm-project/llvm/test/Transforms/LoopVectorize/RISCV/reg-usage-bf16.ll (revision c836b8956d393f98e0d4e136799a33f1bd06e5f5)
1; REQUIRES: asserts
2; RUN: opt -passes=loop-vectorize -mtriple riscv64 -mattr=+v,+zvfbfmin -debug-only=loop-vectorize --disable-output -riscv-v-register-bit-width-lmul=1 -S < %s 2>&1 | FileCheck %s
3
4define void @add(ptr noalias nocapture readonly %src1, ptr noalias nocapture readonly %src2, i32 signext %size, ptr noalias nocapture writeonly %result) {
5; CHECK-LABEL: add
6; CHECK:       LV(REG): Found max usage: 2 item
7; CHECK-NEXT:  LV(REG): RegisterClass: RISCV::GPRRC, 2 registers
8; CHECK-NEXT:  LV(REG): RegisterClass: RISCV::VRRC, 4 registers
9; CHECK-NEXT:  LV(REG): Found invariant usage: 1 item
10; CHECK-NEXT:  LV(REG): RegisterClass: RISCV::GPRRC, 1 registers
11
12entry:
13  %conv = zext i32 %size to i64
14  %cmp10.not = icmp eq i32 %size, 0
15  br i1 %cmp10.not, label %for.cond.cleanup, label %for.body
16
17for.cond.cleanup:
18  ret void
19
20for.body:
21  %i.011 = phi i64 [ %add4, %for.body ], [ 0, %entry ]
22  %arrayidx = getelementptr inbounds bfloat, ptr %src1, i64 %i.011
23  %0 = load bfloat, ptr %arrayidx, align 4
24  %arrayidx2 = getelementptr inbounds bfloat, ptr %src2, i64 %i.011
25  %1 = load bfloat, ptr %arrayidx2, align 4
26  %add = fadd bfloat %0, %1
27  %arrayidx3 = getelementptr inbounds bfloat, ptr %result, i64 %i.011
28  store bfloat %add, ptr %arrayidx3, align 4
29  %add4 = add nuw nsw i64 %i.011, 1
30  %exitcond.not = icmp eq i64 %add4, %conv
31  br i1 %exitcond.not, label %for.cond.cleanup, label %for.body
32}
33