update

Author: deependujha

tests/tests_pytorch/strategies/test_fsdp.py

@@ -138,6 +138,8 @@ def _assert_layer_fsdp_instance(self) -> None:
             param_dtype = reduce_dtype = buffer_dtype = torch.float16
         elif self.trainer.precision in ("bf16-true", "bf16-mixed"):
             param_dtype = reduce_dtype = buffer_dtype = torch.bfloat16
+        elif self.trainer.precision == "32-true":
+            param_dtype = reduce_dtype = buffer_dtype = torch.float32
         else:
             raise ValueError(f"Unknown precision {self.trainer.precision}")