1 2 3 4 5 6 7 8 9 10
#include <clc/clc.h> _CLC_DEF size_t get_num_groups(uint dim) { switch (dim) { case 0: return __builtin_ptx_read_nctaid_x(); case 1: return __builtin_ptx_read_nctaid_y(); case 2: return __builtin_ptx_read_nctaid_z(); default: return 0; } }