blob: f0e52f1fdbc0888916b5c7f09b43b59c5714a7cf (
plain)
1
2
3
4
5
6
7
8
9
10
|
#include <clc/clc.h>
_CLC_DEF _CLC_OVERLOAD size_t get_num_groups(uint dim) {
switch (dim) {
case 0: return __nvvm_read_ptx_sreg_nctaid_x();
case 1: return __nvvm_read_ptx_sreg_nctaid_y();
case 2: return __nvvm_read_ptx_sreg_nctaid_z();
default: return 0;
}
}
|