Added an context for controlling io about intermediate reduction state