[test only] testing adding optioanl tensor arg to float8 tensor

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

stack-info: PR: #2840, branch: jerryzh168/stack/33

Author: jerryzh168

torchao/quantization/quantize_/workflows/float8/float8_tensor.py

@@ -95,6 +95,7 @@ class Float8Tensor(TorchAOBaseTensor):
 
     tensor_data_names = ["qdata", "scale"]
     tensor_attribute_names = []
+    optional_tensor_data_names = ["test_only_data"]
     optional_tensor_attribute_names = [
         "block_size",
         "mm_config",
@@ -109,6 +110,7 @@ def __new__(
         cls,
         qdata: torch.Tensor,
         scale: torch.Tensor,
+        test_only_data: Optional[torch.Tensor] = None,
         block_size: Optional[List[int]] = None,
         mm_config: Optional[Float8MMConfig] = None,
         hp_value_lb: Optional[float] = None,
@@ -128,6 +130,7 @@ def __init__(
         self,
         qdata: torch.Tensor,
         scale: torch.Tensor,
+        test_only_data: Optional[torch.Tensor] = None,
         block_size: Optional[List[int]] = None,
         mm_config: Optional[Float8MMConfig] = None,
         hp_value_lb: Optional[float] = None,
@@ -138,6 +141,7 @@ def __init__(
     ):
         self.qdata = qdata
         self.scale = scale
+        self.test_only_data = test_only_data
         self.block_size = block_size
         self.mm_config = mm_config
         self.hp_value_lb = hp_value_lb