Update tensor.py

pytorch · Sep 23, 2024 · 5490b37 · 5490b37
1 parent 7c6ca1d
commit 5490b37
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/experimental/torch_xla2/torch_xla2/tensor.py b/experimental/torch_xla2/torch_xla2/tensor.py
@@ -205,6 +205,10 @@ def __torch_function__(self,
         return self.env.dispatch(func, types, args, kwargs)
       except OperatorNotFound:
         pass
+      if _name_of_func(func) in ('rot90'): # skip rot90 with k%4==0 due to no change
+        if len(args) >= 2 and type(args[1]) == int:
+          if ((args[1])%4 == 0):
+            return args[0]
       return func(*args, **(kwargs or {}))