| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | ApplyGridUtils.cuh | 23 …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()
|
| H A D | CUDAApplyUtils.cuh | 359 C10_LAUNCH_BOUNDS_2(max_threads_per_block, min_blocks_per_sm) in C10_LAUNCH_BOUNDS_2() argument
|
| /aosp_15_r20/external/clang/test/CodeGenCUDA/ |
| H A D | launch-bounds.cu | 34 __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
|
| /aosp_15_r20/external/pytorch/c10/macros/ |
| H A D | Macros.h | 314 #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
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/softpipe/ |
| H A D | sp_screen.c | 538 uint64_t *max_threads_per_block = ret; in softpipe_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/clover/core/ |
| H A D | device.cpp | 315 device::max_threads_per_block() const { in max_threads_per_block() function in device
|
| /aosp_15_r20/external/mesa3d/src/mesa/state_tracker/ |
| H A D | st_extensions.c | 1690 uint64_t max_local_size, max_threads_per_block; in st_init_extensions() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
| H A D | lp_screen.c | 515 uint64_t *max_threads_per_block = ret; in llvmpipe_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/ |
| H A D | si_get.c | 1251 uint64_t *max_threads_per_block = ret; in si_get_compute_param() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A D | cupti_collector.cc | 409 auto max_threads_per_block = in GetDeviceCapabilities() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/virgl/ |
| H A D | virgl_screen.c | 645 uint64_t *max_threads_per_block = ret; in virgl_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/r600/ |
| H A D | r600_pipe_common.c | 941 uint64_t *max_threads_per_block = ret; in r600_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/core/ |
| H A D | device.rs | 953 pub fn max_threads_per_block(&self) -> usize { in max_threads_per_block() method
|
| H A D | kernel.rs | 1703 pub fn max_threads_per_block(&self, dev: &Device) -> usize { in max_threads_per_block() method
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/lavapipe/ |
| H A D | lvp_device.c | 776 uint64_t max_threads_per_block, max_local_size; in lvp_get_properties() local
|
| /aosp_15_r20/external/mesa3d/src/amd/vulkan/ |
| H A D | radv_device.c | 1192 uint32_t max_threads_per_block = 2048; in radv_CreateDevice() local
|