Cocktailer: Analyzing and Optimizing Dynamic Control Flow in Deep Learning
summary
Cocktailer is a DNN compiler that targets models with dynamic control flow (loops, branches, recursion) and aims to run both control and data flow on accelerators like GPUs instead of handling control on the CPU. It introduces a uTask abstraction to represent control and data uniformly, exposes a global scheduling space to map control flow onto accelerator parallelism, and uses a heuristic scheduler to move control into accelerator kernels for cross-boundary optimizations, claiming up to 8.2× speedups over current frameworks.
AI-written summary · about
related talks
14mCocktail: A Multidimensional Optimization for Model Serving in Cloud
25mDaydream: Accurately Estimating the Efficacy of Optimizations for DNN Training
27mA Profile-Driven Dynamic Optimization Framework for GPU Applications
22mBeacons: An End-to-End Compiler Framework for Predicting and Utilizing Dynamic ...
11mAccelerated Software for a Post-Moore World
21mCe Zhang, Postdoctoral Researcher, Stanford University
37mMonadic Deep Learning
36mDL Compiler Panel Discussion
