Searched refs:real_tasks (Results 1 – 3 of 3) sorted by relevance
67 const int real_tasks = std::max( in PrepareGemmTasks() local71 if (real_tasks == 1) { in PrepareGemmTasks()78 const int m_chunk = params.m / real_tasks; in PrepareGemmTasks()79 for (int i = 0; i < real_tasks - 1; ++i) { in PrepareGemmTasks()84 const int sum_m = (real_tasks - 1) * m_chunk; in PrepareGemmTasks()89 const int n_chunk = params.n / real_tasks; in PrepareGemmTasks()90 for (int i = 0; i < real_tasks - 1; ++i) { in PrepareGemmTasks()95 int sum_n = (real_tasks - 1) * n_chunk; in PrepareGemmTasks()
90 std::int32_t real_tasks = std::max( in PrepareTasks() local93 if (real_tasks == 1) { in PrepareTasks()99 const std::int32_t m_chunk = m / real_tasks; in PrepareTasks()100 for (int i = 0; i < real_tasks - 1; ++i) { in PrepareTasks()103 const std::int32_t last_m_offset = (real_tasks - 1) * m_chunk; in PrepareTasks()106 const std::int32_t n_chunk = n / real_tasks; in PrepareTasks()107 for (int i = 0; i < real_tasks - 1; ++i) { in PrepareTasks()110 const std::int32_t last_n_offset = (real_tasks - 1) * n_chunk; in PrepareTasks()
41 const int real_tasks = std::max(1, std::min(max_threads, max_tasks_by_size)); in PrepareTransform1DTasks() local43 if (real_tasks == 1) { in PrepareTransform1DTasks()47 const int chunk = params.kernel.count / real_tasks; in PrepareTransform1DTasks()48 for (int i = 0; i < real_tasks - 1; ++i) { in PrepareTransform1DTasks()57 const int sum_chunk = (real_tasks - 1) * chunk; in PrepareTransform1DTasks()