Return total memory when limit is max value. (#2214)

Return total memory when limit is max value. (#2214) #443