def group_index(self, env):
"""
group_index()
3-Dimensional index of the block within the launched grid.
"""
from cupyx.jit._interface import _Dim3 # avoid circular import
self._check_cg_include(env)
return _Data('group_index()', _Dim3())
def thread_index(self, env):
"""
thread_index()
3-Dimensional index of the thread within the launched block.
"""
from cupyx.jit._interface import _Dim3 # avoid circular import
self._check_cg_include(env)
return _Data('thread_index()', _Dim3())
def group_dim(self, env):
"""
group_dim()
Dimensions of the launched block in units of threads.
"""
# despite it is an alias of dim_threads, we need it for earlier 11.x
from cupyx.jit._interface import _Dim3 # avoid circular import
self._check_cg_include(env)
return _Data('group_dim()', _Dim3())
def dim_threads(self, env):
"""
dim_threads()
Dimensions of the launched block in units of threads.
"""
if _runtime.runtimeGetVersion() < 11060:
raise RuntimeError("dim_threads() is supported on CUDA 11.6+")
from cupyx.jit._interface import _Dim3 # avoid circular import
self._check_cg_include(env)
return _Data('dim_threads()', _Dim3())
def block_index(self, env):
"""
block_index()
3-Dimensional index of the block within the launched grid.
"""
if _runtime.runtimeGetVersion() < 11060:
raise RuntimeError("block_index() is supported on CUDA 11.6+")
from cupyx.jit._interface import _Dim3 # avoid circular import
self._check_cg_include(env)
return _Data('block_index()', _Dim3())