Sign in
llvm
/
llvm-project
/
aa5ec34e312699a2fe5263133129a1e103ac91ee
/
.
/
libclc
/
ptx-nvidiacl
/
lib
/
workitem
/
get_group_id.cl
blob: dbc47847f9e37e8d750fa4f69d078f9c1ac1fac9 [
file
] [
log
] [
blame
]
#
include
<
clc
/
clc
.
h
>
_CLC_DEF size_t get_group_id
(
uint dim
)
{
switch
(
dim
)
{
case
0
:
return __nvvm_read_ptx_sreg_ctaid_x
(
)
;
case
1
:
return __nvvm_read_ptx_sreg_ctaid_y
(
)
;
case
2
:
return __nvvm_read_ptx_sreg_ctaid_z
(
)
;
default
:
return
0
;
}
}