test(mfsdp): cover v2 training memory (#6567)