[Checkpoint][Test] Add 2d DCP model state checkpoint test (save/load) (#91046)
Add test to test 2D checkpoint save/load functionality for model state.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91046
Approved by: https://github.com/fduwjj