def METHOD_NAME(self):
    """
    Tests that the post-backward reshard happens even for flat parameters
    that do not require gradients.
    """
    self.run_subtests(
        {
            "sharding_strategy": [
                ShardingStrategy.FULL_SHARD,
                ShardingStrategy.SHARD_GRAD_OP,
                ShardingStrategy.NO_SHARD,
            ],
            "use_orig_params": [False, True],
            "inp_requires_grad": [False, True],
            "unfreeze_params": [False, True],
        },
        self._test_backward_reshard_hooks,
    )