Sign in
llvm
/
llvm-project
/
eb1c2bdc1f55fbc5d1e7bb86e9f0e038b0f5adb7
/
.
/
libclc
/
ptx-nvidiacl
/
include
/
clc
/
workitem
/
get_num_groups.h
blob: 36ee849c613f6cb4f1b4f4be63a31e2a412e1c73 [
file
] [
log
] [
blame
]
_CLC_INLINE
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
;
}
}