forked from OSchip/llvm-project
11 lines
256 B
Common Lisp
11 lines
256 B
Common Lisp
|
#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;
|
||
|
|
}
|
||
|
|
}
|