def METHOD_NAME(
    self, input: Dict[nn.Module, torch.Tensor], fsdp_model: nn.Module
):
    # validate eval mode always forces full precision
    fsdp_model.eval()
    _ = fsdp_model(input)
    self.assertEqual(fsdp_model.forward_inputs[fsdp_model.c1].dtype, torch.float32)
    self.assertEqual(fsdp_model.forward_inputs[fsdp_model.c2].dtype, torch.float32)
    fsdp_model.train()