Unverified Commit 48b8eea6 authored by Kirthi Shankar Sivamani's avatar Kirthi Shankar Sivamani Committed by GitHub
Browse files

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



Don't set data to null
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent de06a34c
......@@ -105,8 +105,6 @@ class Float8TensorBase:
"""
tensors = [self._data, self._transpose]
self._data = None
self._transpose = None
return tensors, self
def restore_from_saved(
......
......@@ -100,8 +100,6 @@ class MXFP8TensorBase:
"""
tensors = [self._rowwise_data, self._columnwise_data]
self._rowwise_data = None
self._columnwise_data = None
return tensors, self
def restore_from_saved(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment