Home
last modified time | relevance | path

Searched refs:max_threads_codegen (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.h121 int max_threads_codegen);
129 int max_threads_codegen);
H A Djit_utils.cpp1310 int max_threads_codegen) { in generate_reduction_code() argument
1325 max_threads_codegen in generate_reduction_code()
1340 int max_threads_codegen) { in generate_reduction_code() argument
1349 env.s("max_threads_lb", std::to_string(max_threads_codegen)); in generate_reduction_code()
H A DReduce.cuh923 int max_threads_codegen = in launch_jitted_reduce_kernel() local
926 desc, vt0, true, false, config.output_vec_size, max_threads_codegen); in launch_jitted_reduce_kernel()