tgsi: Add get_group_id / get_local_size / get_num_groups support

Signed-off-by: Hans de Goede <hdegoede@redhat.com>

4 files changed, 57 insertions, 0 deletions

diff --git a/tgsi/lib/SOURCES b/tgsi/lib/SOURCES
index c116b86..4aba61d 100644
--- a/tgsi/lib/SOURCES
+++ b/tgsi/lib/SOURCES
@@ -1 +1,4 @@
+workitem/get_group_id.ll
 workitem/get_local_id.ll
+workitem/get_local_size.ll
+workitem/get_num_groups.ll
diff --git a/tgsi/lib/workitem/get_group_id.ll b/tgsi/lib/workitem/get_group_id.ll
new file mode 100644
index 0000000..f0809af
--- /dev/null
+++ b/tgsi/lib/workitem/get_group_id.ll
@@ -0,0 +1,18 @@
+declare i32 @llvm.tgsi.read.blockid.x() nounwind readnone
+declare i32 @llvm.tgsi.read.blockid.y() nounwind readnone
+declare i32 @llvm.tgsi.read.blockid.z() nounwind readnone
+
+define i32 @get_group_id(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.blockid.x() nounwind readnone
+  ret i32 %x
+y_dim:
+  %y = call i32 @llvm.tgsi.read.blockid.y() nounwind readnone
+  ret i32 %y
+z_dim:
+  %z = call i32 @llvm.tgsi.read.blockid.z() nounwind readnone
+  ret i32 %z
+default:
+  ret i32 0
+}
diff --git a/tgsi/lib/workitem/get_local_size.ll b/tgsi/lib/workitem/get_local_size.ll
new file mode 100644
index 0000000..8651dc9
--- /dev/null
+++ b/tgsi/lib/workitem/get_local_size.ll
@@ -0,0 +1,18 @@
+declare i32 @llvm.tgsi.read.blocksize.x() nounwind readnone
+declare i32 @llvm.tgsi.read.blocksize.y() nounwind readnone
+declare i32 @llvm.tgsi.read.blocksize.z() nounwind readnone
+
+define i32 @get_local_size(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.blocksize.x() nounwind readnone
+  ret i32 %x
+y_dim:
+  %y = call i32 @llvm.tgsi.read.blocksize.y() nounwind readnone
+  ret i32 %y
+z_dim:
+  %z = call i32 @llvm.tgsi.read.blocksize.z() nounwind readnone
+  ret i32 %z
+default:
+  ret i32 1
+}
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
+}