Unverified Commit 31fc29ab authored by Tim Moon's avatar Tim Moon Committed by GitHub
Browse files

[PyTorch] Make `MXFP8Tensor` unpickling function backward compatible (#1943)



Make MXFP8Tensor unpickling function backward compatible
Signed-off-by: default avatarTim Moon <tmoon@nvidia.com>
parent 62acae05
...@@ -328,7 +328,7 @@ class MXFP8Tensor(MXFP8TensorBase, QuantizedTensor): ...@@ -328,7 +328,7 @@ class MXFP8Tensor(MXFP8TensorBase, QuantizedTensor):
fp8_dtype: TE_DType, fp8_dtype: TE_DType,
dtype: torch.dtype, dtype: torch.dtype,
shape: torch.shape, shape: torch.shape,
quantizer: Quantizer, quantizer: Optional[Quantizer] = None,
) -> MXFP8Tensor: ) -> MXFP8Tensor:
"""Build MXFP8Tensor, for use in __reduce__ """Build MXFP8Tensor, for use in __reduce__
......
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