From 9ddf3314d80bb362f9e01e4d7c6f089e029a4da0 Mon Sep 17 00:00:00 2001 From: drisspg Date: Tue, 9 Apr 2024 18:36:05 -0700 Subject: [PATCH] feels too easy... --- float8_experimental/float8_tensor.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/float8_experimental/float8_tensor.py b/float8_experimental/float8_tensor.py index 7287c11..27117d2 100644 --- a/float8_experimental/float8_tensor.py +++ b/float8_experimental/float8_tensor.py @@ -325,6 +325,8 @@ def allowed_subclasses(type): if func in FLOAT8_OPS_TABLE: return FLOAT8_OPS_TABLE[func](func, args, kwargs) + else: + return func.decompose(*args, *kwargs) raise NotImplementedError(f"attempting to run {func}, this is not supported") # Do not force the Float8Tensor type on the returned tensor