update wake memory allocation using MaxTimeSteps