libclc/ptx-nvidiacl/lib/workitem/get_local_id.cl


1
2
3
4
5
6
7
8
9
10

#include <clc/clc.h>

_CLC_DEF size_t get_local_id(uint dim) {
  switch (dim) {
  case 0:  return __nvvm_read_ptx_sreg_tid_x();
  case 1:  return __nvvm_read_ptx_sreg_tid_y();
  case 2:  return __nvvm_read_ptx_sreg_tid_z();
  default: return 0;
  }
}