Searched refs:max_num_threads_per_block (Results 1 – 2 of 2) sorted by relevance
63 if (numel_per_split <= max_num_threads_per_block * numel_per_thread) { in init_elementwise_launch_config()71 numel_per_split, max_num_threads_per_block * numel_per_thread), in init_elementwise_launch_config()73 num_threads = max_num_threads_per_block; in init_elementwise_launch_config()
11 constexpr size_t max_num_threads_per_block = 1024; variable