In the memory profile, print the max memory used on the GPU.
Updated a little how the memory profile get printed.