parallelize

CPU implementations of parallelization functions.

Functions

• ​func_unified:
• ​parallelize: Executes func(0) ... func(num_work_items-1) as sub-tasks in parallel, and returns when all are complete.
• ​parallelize_over_rows: Parallelize func over non-axis dims of shape.
• ​sync_parallelize: Executes func(0) ... func(num_work_items-1) as parallel sub-tasks, and returns when all are complete.