Sign in
llvm
/
libclc
/
9f6204ec04a8cadb6bef57caa71e3161c4f398f2
/
.
/
ptx-nvidiacl
/
lib
/
workitem
/
get_local_size.cl
blob: d00b0d6c9fba7f1fe81208812a6ce7d4016fd652 [
file
] [
log
] [
blame
]
#
include
<
clc
/
clc
.
h
>
_CLC_DEF size_t get_local_size
(
uint dim
)
{
switch
(
dim
)
{
case
0
:
return __nvvm_read_ptx_sreg_ntid_x
(
)
;
case
1
:
return __nvvm_read_ptx_sreg_ntid_y
(
)
;
case
2
:
return __nvvm_read_ptx_sreg_ntid_z
(
)
;
default
:
return
0
;
}
}