llvm-project/libclc/r600/lib/workitem/get_num_groups.ll

21 lines
749 B
LLVM

declare i32 @llvm.r600.read.ngroups.x() nounwind readnone
declare i32 @llvm.r600.read.ngroups.y() nounwind readnone
declare i32 @llvm.r600.read.ngroups.z() nounwind readnone
target datalayout = "e-p:32:32-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64-S32-A5"
define i32 @get_num_groups(i32 %dim) nounwind readnone alwaysinline {
switch i32 %dim, label %default [i32 0, label %x_dim i32 1, label %y_dim i32 2, label %z_dim]
x_dim:
%x = call i32 @llvm.r600.read.ngroups.x() nounwind readnone
ret i32 %x
y_dim:
%y = call i32 @llvm.r600.read.ngroups.y() nounwind readnone
ret i32 %y
z_dim:
%z = call i32 @llvm.r600.read.ngroups.z() nounwind readnone
ret i32 %z
default:
ret i32 1
}