1 files changed, 8 insertions, 0 deletions
diff --git a/ptx-nvidiacl/include/clc/workitem/get_group_id.h b/ptx-nvidiacl/include/clc/workitem/get_group_id.h
new file mode 100644
index 0000000..18b1bd4
--- /dev/null
+++ b/ptx-nvidiacl/include/clc/workitem/get_group_id.h
@@ -0,0 +1,8 @@
+_CLC_INLINE size_t get_group_id(uint dim) {
+  switch (dim) {
+  case 0:  return __builtin_ptx_read_ctaid_x();
+  case 1:  return __builtin_ptx_read_ctaid_y();
+  case 2:  return __builtin_ptx_read_ctaid_z();
+  default: return 0;
+  }
+}