diff options
Diffstat (limited to 'ptx-nvidiacl/lib/workitem/get_local_size.cl')
-rw-r--r-- | ptx-nvidiacl/lib/workitem/get_local_size.cl | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/ptx-nvidiacl/lib/workitem/get_local_size.cl b/ptx-nvidiacl/lib/workitem/get_local_size.cl new file mode 100644 index 0000000..c3f5425 --- /dev/null +++ b/ptx-nvidiacl/lib/workitem/get_local_size.cl @@ -0,0 +1,10 @@ +#include <clc/clc.h> + +_CLC_DEF size_t get_local_size(uint dim) { + switch (dim) { + case 0: return __builtin_ptx_read_ntid_x(); + case 1: return __builtin_ptx_read_ntid_y(); + case 2: return __builtin_ptx_read_ntid_z(); + default: return 0; + } +} |