pytaco.set_parallel_schedule

pytaco.set_parallel_schedule(sched_type, chunk_size)

Sets the strategy for performing computations in parallel.

Parameters

sched_type: string

Either “static” or “dynamic”. “static” indicates that Taco should parallelize subsequent computations using a strategy that assigns the same number of coordinates along a particular dimension to be processed by each thread. “dynamic” indicates that Taco should parallelize subsequent computations using a strategy that assigns work to the threads at runtime for better load balance.

chunk_size: int

For a dynamic schedule, the amount of additional work that is assigned to any idle thread.