[PyTorch] Don't set FP8 data to None when saving base tensors (#1548)

Don't set data to null

Signed-off-by: Kirthi Shankar Sivamani <[email protected]>

Author: ksivaman

transformer_engine/pytorch/tensor/_internal/float8_tensor_base.py

@@ -105,8 +105,6 @@ def prepare_for_saving(self) -> Tuple[list[Optional[torch.Tensor]], Float8Tensor
 
         """
         tensors = [self._data, self._transpose]
-        self._data = None
-        self._transpose = None
         return tensors, self
 
     def restore_from_saved(

transformer_engine/pytorch/tensor/_internal/mxfp8_tensor_base.py

@@ -100,8 +100,6 @@ def prepare_for_saving(self) -> Tuple[list[Optional[torch.Tensor]], MXFP8TensorB
 
         """
         tensors = [self._rowwise_data, self._columnwise_data]
-        self._rowwise_data = None
-        self._columnwise_data = None
         return tensors, self
 
     def restore_from_saved(