forked from OSchip/llvm-project
67 lines
1.8 KiB
LLVM
67 lines
1.8 KiB
LLVM
; RUN: opt -mtriple=amdgcn-- -loop-unroll -simplifycfg -sroa %s -S -o - | FileCheck %s
|
|
; RUN: opt -mtriple=r600-- -loop-unroll -simplifycfg -sroa %s -S -o - | FileCheck %s
|
|
|
|
|
|
; This test contains a simple loop that initializes an array declared in
|
|
; private memory. We want to make sure these kinds of loops are always
|
|
; unrolled, because private memory is slow.
|
|
|
|
; CHECK-LABEL: @private_memory
|
|
; CHECK-NOT: alloca
|
|
; CHECK: store i32 5, i32 addrspace(1)* %out
|
|
define amdgpu_kernel void @private_memory(i32 addrspace(1)* %out) {
|
|
entry:
|
|
%0 = alloca [32 x i32]
|
|
br label %loop.header
|
|
|
|
loop.header:
|
|
%counter = phi i32 [0, %entry], [%inc, %loop.inc]
|
|
br label %loop.body
|
|
|
|
loop.body:
|
|
%ptr = getelementptr [32 x i32], [32 x i32]* %0, i32 0, i32 %counter
|
|
store i32 %counter, i32* %ptr
|
|
br label %loop.inc
|
|
|
|
loop.inc:
|
|
%inc = add i32 %counter, 1
|
|
%1 = icmp sge i32 %counter, 32
|
|
br i1 %1, label %exit, label %loop.header
|
|
|
|
exit:
|
|
%2 = getelementptr [32 x i32], [32 x i32]* %0, i32 0, i32 5
|
|
%3 = load i32, i32* %2
|
|
store i32 %3, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; Check that loop is unrolled for local memory references
|
|
|
|
; CHECK-LABEL: @local_memory
|
|
; CHECK: getelementptr i32, i32 addrspace(1)* %out, i32 128
|
|
; CHECK-NEXT: store
|
|
; CHECK-NEXT: ret
|
|
define amdgpu_kernel void @local_memory(i32 addrspace(1)* %out, i32 addrspace(3)* %lds) {
|
|
entry:
|
|
br label %loop.header
|
|
|
|
loop.header:
|
|
%counter = phi i32 [0, %entry], [%inc, %loop.inc]
|
|
br label %loop.body
|
|
|
|
loop.body:
|
|
%ptr_lds = getelementptr i32, i32 addrspace(3)* %lds, i32 %counter
|
|
%val = load i32, i32 addrspace(3)* %ptr_lds
|
|
%ptr_out = getelementptr i32, i32 addrspace(1)* %out, i32 %counter
|
|
store i32 %val, i32 addrspace(1)* %ptr_out
|
|
br label %loop.inc
|
|
|
|
loop.inc:
|
|
%inc = add i32 %counter, 1
|
|
%cond = icmp sge i32 %counter, 128
|
|
br i1 %cond, label %exit, label %loop.header
|
|
|
|
exit:
|
|
ret void
|
|
}
|