Home
last modified time | relevance | path

Searched refs:threads_per_threadgroup (Results 1 – 4 of 4) sorted by relevance

/external/mesa3d/src/amd/vulkan/
Dradv_pipeline_compute.c44 unsigned threads_per_threadgroup; in radv_get_compute_resource_limits() local
50threads_per_threadgroup = info->cs.block_size[0] * info->cs.block_size[1] * info->cs.block_size[2]; in radv_get_compute_resource_limits()
51 waves_per_threadgroup = DIV_ROUND_UP(threads_per_threadgroup, info->wave_size); in radv_get_compute_resource_limits()
/external/pytorch/aten/src/ATen/native/mps/operations/
DBucketization.mm106 uint2 tptg [[threads_per_threadgroup]]) {
134 uint2 tptg [[threads_per_threadgroup]]) {
165 uint2 tptg [[threads_per_threadgroup]]); \
179 uint2 tptg [[threads_per_threadgroup]]); \
DRepeat.mm92 … uint threads_per_threadgroup [[threads_per_threadgroup]],
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_compute.c887 unsigned threads_per_threadgroup = info->block[0] * info->block[1] * info->block[2]; in si_get_2d_interleave_size() local
888 unsigned workgroups_per_se = MIN2(max_threads_per_se / threads_per_threadgroup, 16); in si_get_2d_interleave_size()
914 unsigned threads_per_threadgroup = info->block[0] * info->block[1] * info->block[2]; in si_emit_dispatch_packets() local
916 DIV_ROUND_UP(threads_per_threadgroup, sctx->cs_shader_state.program->shader.wave_size); in si_emit_dispatch_packets()