revert change to scale dtype as it seems to work now

pytorch-labs · Jun 20, 2024 · 35a5dfc · 35a5dfc
1 parent eec175a
commit 35a5dfc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/float8_experimental/float8_ops.py b/float8_experimental/float8_ops.py
@@ -44,7 +44,6 @@ def decorator(func):
         aten.slice.Tensor,
         aten.transpose.int,
         aten.fill_.Scalar,
-        aten.copy_.default,
     ]
 )
 def float8_desugar_op(aten_op, args, kwargs=None):

diff --git a/float8_experimental/float8_tensor.py b/float8_experimental/float8_tensor.py
@@ -157,7 +157,7 @@ class FromFloat8ConstrFunc(torch.autograd.Function):
 
     @staticmethod
     def forward(ctx, tensor):
-        return (tensor._data.to(tensor._orig_dtype) / tensor._scale.to(tensor._org_dtype))
+        return tensor._data.to(tensor._orig_dtype) / tensor._scale
 
     @staticmethod
     def backward(ctx, g):