Home
last modified time | relevance | path

Searched defs:max_threads_per_block (Results 1 – 16 of 16) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DApplyGridUtils.cuh23 …uint64_t totalElements, dim3& grid, c10::DeviceIndex curDevice, int max_threads_per_block=AT_APPLY… in getApplyGrid()
42 inline dim3 getApplyBlock(int max_threads_per_block=AT_APPLY_THREADS_PER_BLOCK) { in getApplyBlock()
DCUDAApplyUtils.cuh359 C10_LAUNCH_BOUNDS_2(max_threads_per_block, min_blocks_per_sm) in C10_LAUNCH_BOUNDS_2() argument
/external/clang/test/CodeGenCUDA/
Dlaunch-bounds.cu34 __launch_bounds__(max_threads_per_block) in __launch_bounds__() argument
44 __launch_bounds__(max_threads_per_block, min_blocks_per_mp) in __launch_bounds__() argument
/external/mesa3d/src/gallium/drivers/softpipe/
Dsp_screen.c442 uint64_t *max_threads_per_block = ret; in softpipe_get_compute_param() local
/external/pytorch/c10/macros/
DMacros.h314 #define C10_LAUNCH_BOUNDS_1(max_threads_per_block) \ argument
316 #define C10_LAUNCH_BOUNDS_2(max_threads_per_block, min_blocks_per_sm) \ argument
/external/mesa3d/src/gallium/frontends/clover/core/
Ddevice.cpp301 device::max_threads_per_block() const { in max_threads_per_block() function in device
/external/mesa3d/src/mesa/state_tracker/
Dst_extensions.c1669 uint64_t max_local_size, max_threads_per_block; in st_init_extensions() local
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_screen.c203 uint64_t *max_threads_per_block = ret; in llvmpipe_get_compute_param() local
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_get.c863 uint64_t *max_threads_per_block = ret; in si_get_compute_param() local
/external/tensorflow/tensorflow/core/profiler/backends/gpu/
Dcupti_collector.cc409 auto max_threads_per_block = in GetDeviceCapabilities() local
/external/mesa3d/src/gallium/drivers/virgl/
Dvirgl_screen.c307 uint64_t *max_threads_per_block = ret; in virgl_get_compute_param() local
/external/mesa3d/src/gallium/drivers/r600/
Dr600_pipe_common.c910 uint64_t *max_threads_per_block = ret; in r600_get_compute_param() local
/external/mesa3d/src/gallium/frontends/rusticl/core/
Ddevice.rs1017 pub fn max_threads_per_block(&self) -> usize { in max_threads_per_block() method
Dkernel.rs1724 pub fn max_threads_per_block(&self, dev: &Device) -> usize { in max_threads_per_block() method
/external/mesa3d/src/amd/vulkan/
Dradv_device.c1269 uint32_t max_threads_per_block = 2048; in radv_CreateDevice() local
/external/mesa3d/src/gallium/frontends/lavapipe/
Dlvp_device.c783 uint64_t max_threads_per_block, max_local_size; in lvp_get_properties() local