changed update forward in memory stage