Near Optimal Work-Stealing Tree Scheduler for Highly Irregular Data-Parallel Workloads

• Published in: Languages and Compilers for Parallel Computing pp. 55 - 86
• Main Authors: Prokopec, Aleksandar, Odersky, Martin
• Format: Book Chapter
• Language: English
• Published: Cham Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: Batch Order; Batch Size; Cache Line; Owned State; Workload Distribution
• ISBN: 3319099663; 9783319099668
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-319-09967-5_4

We present a work-stealing algorithm for runtime scheduling of data-parallel operations in the context of shared-memory architectures on data sets with highly-irregular workloads that are not known a priori to the scheduler. This scheduler can parallelize loops and operations expressible with a parallel reduce or a parallel scan. The scheduler is based on the work-stealing tree data structure, which allows workers to decide on the work division in a lock-free, workload-driven manner and attempts to minimize the amount of communication between them. A significant effort is given to showing that the algorithm has the least possible amount of overhead. We provide an extensive experimental evaluation, comparing the advantages and shortcomings of different data-parallel schedulers in order to combine their strengths. We show specific workload distribution patterns appearing in practice for which different schedulers yield suboptimal speedup, explaining their drawbacks and demonstrating how the work-stealing tree scheduler overcomes them. We thus justify our design decisions experimentally, but also provide a theoretical background for our claims.