Mojo function
get_matmul_num_tasks
get_matmul_num_tasks[a_type: DType, b_type: DType, c_type: DType, simd_size: Int, kernel_type: Bool](m: Int, n: Int, k: Int, max_num_tasks: Int) -> Int
Compute the number of tasks for parallel matmul. The max number of tasks is typically the number of threads/cores.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!