Sign in
llvm
/
llvm-project
/
libclc
/
42601e3c3aebbe89ce0337d425d4aeed936d4e24
/
.
/
ptx-nvidiacl
/
lib
/
workitem
/
get_num_groups.cl
blob: d7abf3f290704fc0063a4ecd3d184f812b24e057 [
file
] [
log
] [
blame
]
#
include
<
clc
/
clc
.
h
>
_CLC_DEF 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
;
}
}