updated DBN IR worker that handles the straggler split sub-batch edge case