add "_gemm_input_role" to dunder slots (#984)

* add "_gemm_input_role" to dunder slots

Signed-off-by: Masaki Kozuki <[email protected]>

* lint

---------

Signed-off-by: Masaki Kozuki <[email protected]>
Co-authored-by: Mark Saroufim <[email protected]>

Author: crcrpar

torchao/float8/float8_tensor.py

@@ -293,8 +293,16 @@ class Float8Tensor(torch.Tensor):
     _scale: torch.Tensor
     _orig_dtype: torch.dtype
     _linear_mm_config: LinearMMConfig
+    _gemm_input_role: GemmInputRole
     _axiswise_dim: Optional[int]
-    __slots__ = ["_data", "_scale", "_orig_dtype", "_linear_mm_config"]
+    __slots__ = [
+        "_data",
+        "_scale",
+        "_orig_dtype",
+        "_linear_mm_config",
+        "_gemm_input_role",
+        "_axiswise_dim",
+    ]
 
     def __new__(
         cls,