diff options
Diffstat (limited to 'tgsi/lib/workitem/get_num_groups.ll')
| -rw-r--r-- | tgsi/lib/workitem/get_num_groups.ll | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/tgsi/lib/workitem/get_num_groups.ll b/tgsi/lib/workitem/get_num_groups.ll new file mode 100644 index 0000000..45cac88 --- /dev/null +++ b/tgsi/lib/workitem/get_num_groups.ll @@ -0,0 +1,18 @@ +declare i32 @llvm.tgsi.read.gridsize.x() nounwind readnone +declare i32 @llvm.tgsi.read.gridsize.y() nounwind readnone +declare i32 @llvm.tgsi.read.gridsize.z() nounwind readnone + +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.tgsi.read.gridsize.x() nounwind readnone + ret i32 %x +y_dim: + %y = call i32 @llvm.tgsi.read.gridsize.y() nounwind readnone + ret i32 %y +z_dim: + %z = call i32 @llvm.tgsi.read.gridsize.z() nounwind readnone + ret i32 %z +default: + ret i32 1 +} |