diff options
Diffstat (limited to 'ptx-nvidiacl/lib/workitem/get_group_id.cl')
| -rw-r--r-- | ptx-nvidiacl/lib/workitem/get_group_id.cl | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/ptx-nvidiacl/lib/workitem/get_group_id.cl b/ptx-nvidiacl/lib/workitem/get_group_id.cl new file mode 100644 index 0000000..2b35b4e --- /dev/null +++ b/ptx-nvidiacl/lib/workitem/get_group_id.cl @@ -0,0 +1,10 @@ +#include <clc/clc.h> + +_CLC_DEF 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; + } +} |