Revert "[DCP][Test]Remove broken 2d checkpoint test (#106367)"
This reverts commit d2a9b256f00742b3fd1271ad087fc4e02144aed8.
Reverted https://github.com/pytorch/pytorch/pull/106367 on behalf of https://github.com/jeanschmidt due to Breaking internal builds for diff D48007925 ([comment](https://github.com/pytorch/pytorch/pull/106367#issuecomment-1665901322))