Skip to main content
Log in

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?