add "_gemm_input_role" to dunder slots

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

Author: crcrpar

torchao/float8/float8_tensor.py

@@ -293,8 +293,9 @@ 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,