Whole-stage code generation