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 D | gemv_driver.cpp | 382 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 D | gemm_driver.cpp | 1642 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 D | gemv_driver.cpp | 382 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 D | gemm_driver.cpp | 1642 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 …]
|