moved the queue depth augmentation to the generation stage so that we get the correct depth while scheduling