fix(torch-frontend): Added unsupported dtype for `torch.nn.functional…

….linear` (#25966)
ivy-llc · Sep 23, 2023 · c956029 · c956029
1 parent 1cee783
commit c956029
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/ivy/functional/backends/paddle/linear_algebra.py b/ivy/functional/backends/paddle/linear_algebra.py
@@ -227,6 +227,7 @@ def matmul(
         paddle.uint8,
         paddle.float16,
         paddle.bool,
+        paddle.bfloat16,
     ]:
         x1, x2 = x1.cast("float32"), x2.cast("float32")
 

diff --git a/ivy/functional/frontends/torch/nn/functional/linear_functions.py b/ivy/functional/frontends/torch/nn/functional/linear_functions.py
@@ -1,8 +1,10 @@
 # local
 import ivy
+from ivy.func_wrapper import with_unsupported_dtypes
 from ivy.functional.frontends.torch.func_wrapper import to_ivy_arrays_and_back
 
 
+@with_unsupported_dtypes({"2.0.1 and below": ("float16",)}, "torch")
 @to_ivy_arrays_and_back
 def linear(input, weight, bias=None):
     return ivy.linear(input, weight, bias=bias)