[Checkpoint Wrapper] Fix assert (#80283)
https://github.com/pytorch/pytorch/blob/f11cce309bf290151a21b2ac8e79bf2a24dcf2f3/test/distributed/fsdp/test_checkpoint_wrapper.py#L103-L109
The second assert is re-asserting the same thing as the first assert, so `functional_reentrant` and `wrapper_reentrant` are unused. This PR fixes the second assert to assert that `functional_reentrant` and `wrapper_reentrant` are equal.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80283
Approved by: https://github.com/rohan-varma
