Home
last modified time | relevance | path

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

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/
H A Dgemv_driver.cpp382 auto nthr_goal = thread_checker<a_t>(nthr_max, m, n, trans); in gemv_threading_driver() local
384 if (nthr_goal == 1) { in gemv_threading_driver()
391 bool is_short_fat = m <= nthr_goal * M_MIN && n >= nthr_goal * N_MIN; in gemv_threading_driver()
398 ybuf = (c_t *)malloc(sizeof(*ybuf) * m * (nthr_goal - 1), PAGE_4K); in gemv_threading_driver()
402 auto nthr_spawn = dnnl_thr_syncable() ? nthr_max : nthr_goal; in gemv_threading_driver()
405 int nthr_eff = nstl::min(nthr_goal, nthr); in gemv_threading_driver()
H A Dgemm_driver.cpp1642 int nthr_goal = nthr_max; in gemm_threading_driver() local
1680 nthr_goal = force_threading->nthrs(); in gemm_threading_driver()
1694 nthr_goal = set_thread_opts(nthr_goal, nthr_max, thread_info, arg); in gemm_threading_driver()
1698 for (int ithr = 0; ithr < nthr_goal; ithr++) { in gemm_threading_driver()
1729 if (nocopy_checker(nthr_goal, arg)) in gemm_threading_driver()
1730 return call_no_copy_sgemm(nthr_goal, arg); in gemm_threading_driver()
1732 if (nthr_goal == 1) in gemm_threading_driver()
1782 for (int ithr = 0; ithr < nthr_goal; ithr++) { in gemm_threading_driver()
1795 int nthr_eff = force_threading ? nthr_goal : nstl::min(nthr_goal, nthr); in gemm_threading_driver()
1906 parallel(nthr_goal, [&](int ithr, int nthr) { in gemm_threading_driver()
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/
H A Dgemv_driver.cpp382 auto nthr_goal = thread_checker<a_t>(nthr_max, m, n, trans); in gemv_threading_driver() local
384 if (nthr_goal == 1) { in gemv_threading_driver()
391 bool is_short_fat = m <= nthr_goal * M_MIN && n >= nthr_goal * N_MIN; in gemv_threading_driver()
398 ybuf = (c_t *)malloc(sizeof(*ybuf) * m * (nthr_goal - 1), PAGE_4K); in gemv_threading_driver()
402 auto nthr_spawn = dnnl_thr_syncable() ? nthr_max : nthr_goal; in gemv_threading_driver()
405 int nthr_eff = nstl::min(nthr_goal, nthr); in gemv_threading_driver()
H A Dgemm_driver.cpp1642 int nthr_goal = nthr_max; in gemm_threading_driver() local
1680 nthr_goal = force_threading->nthrs(); in gemm_threading_driver()
1694 nthr_goal = set_thread_opts(nthr_goal, nthr_max, thread_info, arg); in gemm_threading_driver()
1698 for (int ithr = 0; ithr < nthr_goal; ithr++) { in gemm_threading_driver()
1729 if (nocopy_checker(nthr_goal, arg)) in gemm_threading_driver()
1730 return call_no_copy_sgemm(nthr_goal, arg); in gemm_threading_driver()
1732 if (nthr_goal == 1) in gemm_threading_driver()
1782 for (int ithr = 0; ithr < nthr_goal; ithr++) { in gemm_threading_driver()
1795 int nthr_eff = force_threading ? nthr_goal : nstl::min(nthr_goal, nthr); in gemm_threading_driver()
1906 parallel(nthr_goal, [&](int ithr, int nthr) { in gemm_threading_driver()
[all …]