0e11a92447
mplement printing for VPIRMetadata, using generic dyn_cast to VPIRMetadata. Depends on https://github.com/llvm/llvm-project/pull/166245 PR: https://github.com/llvm/llvm-project/pull/168385
113 lines
4.4 KiB
LLVM
113 lines
4.4 KiB
LLVM
; REQUIRES: asserts
|
|
|
|
; RUN: opt -passes=loop-vectorize -debug-only=loop-vectorize -force-vector-interleave=1 -force-vector-width=4 -disable-output %s 2>&1 | FileCheck %s
|
|
|
|
define void @test_widen_metadata(ptr noalias %A, ptr noalias %B, i32 %n) {
|
|
; CHECK-LABEL: Checking a loop in 'test_widen_metadata'
|
|
; CHECK: VPlan 'Initial VPlan for VF={4},UF>=1' {
|
|
; CHECK: <x1> vector loop: {
|
|
; CHECK: vector.body:
|
|
; CHECK: WIDEN ir<%lv> = load vp<{{.*}}> (!tbaa ![[TBAA:[0-9]+]])
|
|
; CHECK: WIDEN-CAST ir<%conv> = sitofp ir<%lv> to float (!fpmath ![[FPMATH:[0-9]+]])
|
|
; CHECK: WIDEN ir<%mul> = fmul ir<%conv>, ir<2.000000e+00> (!fpmath ![[FPMATH]])
|
|
; CHECK: WIDEN-CAST ir<%conv.back> = fptosi ir<%mul> to i32
|
|
; CHECK: WIDEN store vp<{{.*}}>, ir<%conv.back> (!tbaa ![[TBAA]])
|
|
; CHECK: ir-bb<loop>:
|
|
; CHECK: IR %lv = load i32, ptr %gep.A, align 4, !tbaa ![[TBAA]]
|
|
; CHECK: IR %conv = sitofp i32 %lv to float, !fpmath ![[FPMATH]]
|
|
; CHECK: IR %mul = fmul float %conv, 2.000000e+00, !fpmath ![[FPMATH]]
|
|
; CHECK: IR store i32 %conv.back, ptr %gep.B, align 4, !tbaa ![[TBAA]]
|
|
;
|
|
entry:
|
|
br label %loop
|
|
|
|
loop:
|
|
%i = phi i32 [ 0, %entry ], [ %i.next, %loop ]
|
|
%gep.A = getelementptr inbounds i32, ptr %A, i32 %i
|
|
%lv = load i32, ptr %gep.A, align 4, !tbaa !0, !range !6
|
|
%conv = sitofp i32 %lv to float, !fpmath !5
|
|
%mul = fmul float %conv, 2.0, !fpmath !5
|
|
%conv.back = fptosi float %mul to i32
|
|
%gep.B = getelementptr inbounds i32, ptr %B, i32 %i
|
|
store i32 %conv.back, ptr %gep.B, align 4, !tbaa !0
|
|
%i.next = add i32 %i, 1
|
|
%cond = icmp eq i32 %i.next, %n
|
|
br i1 %cond, label %exit, label %loop
|
|
|
|
exit:
|
|
ret void
|
|
}
|
|
|
|
declare float @llvm.sqrt.f32(float)
|
|
|
|
define void @test_intrinsic_with_metadata(ptr noalias %A, ptr noalias %B, i32 %n) {
|
|
; CHECK-LABEL: Checking a loop in 'test_intrinsic_with_metadata'
|
|
; CHECK: VPlan 'Initial VPlan for VF={4},UF>=1' {
|
|
; CHECK: <x1> vector loop: {
|
|
; CHECK: vector.body:
|
|
; CHECK: WIDEN ir<%lv> = load vp<{{.*}}> (!tbaa ![[TBAA2:[0-9]+]])
|
|
; CHECK: WIDEN-INTRINSIC ir<%sqrt> = call llvm.sqrt(ir<%lv>) (!fpmath ![[FPMATH2:[0-9]+]])
|
|
; CHECK: WIDEN store vp<{{.*}}>, ir<%sqrt> (!tbaa ![[TBAA2]])
|
|
; CHECK: ir-bb<loop>:
|
|
; CHECK: IR %lv = load float, ptr %gep.A, align 4, !tbaa ![[TBAA2]]
|
|
; CHECK: IR %sqrt = call float @llvm.sqrt.f32(float %lv), !fpmath ![[FPMATH2]]
|
|
; CHECK: IR store float %sqrt, ptr %gep.B, align 4, !tbaa ![[TBAA2]]
|
|
;
|
|
entry:
|
|
br label %loop
|
|
|
|
loop:
|
|
%i = phi i32 [ 0, %entry ], [ %i.next, %loop ]
|
|
%gep.A = getelementptr inbounds float, ptr %A, i32 %i
|
|
%lv = load float, ptr %gep.A, align 4, !tbaa !0
|
|
%sqrt = call float @llvm.sqrt.f32(float %lv), !fpmath !5
|
|
%gep.B = getelementptr inbounds float, ptr %B, i32 %i
|
|
store float %sqrt, ptr %gep.B, align 4, !tbaa !0
|
|
%i.next = add i32 %i, 1
|
|
%cond = icmp eq i32 %i.next, %n
|
|
br i1 %cond, label %exit, label %loop
|
|
|
|
exit:
|
|
ret void
|
|
}
|
|
|
|
define void @test_widen_with_multiple_metadata(ptr noalias %A, ptr noalias %B, i32 %n) {
|
|
; CHECK-LABEL: Checking a loop in 'test_widen_with_multiple_metadata'
|
|
; CHECK: VPlan 'Initial VPlan for VF={4},UF>=1' {
|
|
; CHECK: <x1> vector loop: {
|
|
; CHECK: vector.body:
|
|
; CHECK: WIDEN ir<%lv> = load vp<{{.*}}> (!tbaa ![[TBAA3:[0-9]+]])
|
|
; CHECK: WIDEN-CAST ir<%conv> = sitofp ir<%lv> to float
|
|
; CHECK: WIDEN ir<%mul> = fmul ir<%conv>, ir<2.000000e+00>
|
|
; CHECK: WIDEN-CAST ir<%conv.back> = fptosi ir<%mul> to i32
|
|
; CHECK: WIDEN store vp<{{.*}}>, ir<%conv.back> (!tbaa ![[TBAA3]])
|
|
; CHECK: ir-bb<loop>:
|
|
; CHECK: IR %lv = load i32, ptr %gep.A, align 4, !tbaa ![[TBAA3]]
|
|
; CHECK: IR store i32 %conv.back, ptr %gep.B, align 4, !tbaa ![[TBAA3]]
|
|
;
|
|
entry:
|
|
br label %loop
|
|
|
|
loop:
|
|
%i = phi i32 [ 0, %entry ], [ %i.next, %loop ]
|
|
%gep.A = getelementptr inbounds i32, ptr %A, i32 %i
|
|
%lv = load i32, ptr %gep.A, align 4, !tbaa !0, !range !6
|
|
%conv = sitofp i32 %lv to float
|
|
%mul = fmul float %conv, 2.0
|
|
%conv.back = fptosi float %mul to i32
|
|
%gep.B = getelementptr inbounds i32, ptr %B, i32 %i
|
|
store i32 %conv.back, ptr %gep.B, align 4, !tbaa !0
|
|
%i.next = add i32 %i, 1
|
|
%cond = icmp eq i32 %i.next, %n
|
|
br i1 %cond, label %exit, label %loop
|
|
|
|
exit:
|
|
ret void
|
|
}
|
|
|
|
!0 = !{!1, !1, i64 0}
|
|
!1 = !{!"float", !2}
|
|
!2 = !{!"root"}
|
|
!5 = !{float 2.500000e+00}
|
|
!6 = !{i32 0, i32 100}
|