swap order in rsqrt

fixed backward for 0
chenyuxyz · Feb 13, 2025 · f8c89ee · f8c89ee
1 parent 49abc09
commit f8c89ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/test/test_ops.py b/test/test_ops.py
@@ -647,8 +647,7 @@ def test_sqrt(self):
     helper_test_op([()], lambda x: x.sqrt())
   def test_rsqrt(self):
     helper_test_op([(45,65)], lambda x: x.rsqrt())
-    # TODO: fix backward
-    helper_test_op(None, lambda x: x.rsqrt(), vals=[[0.0]], forward_only=True)
+    helper_test_op(None, lambda x: x.rsqrt(), vals=[[0.0]])
     helper_test_op([()], lambda x: x.rsqrt())
 
   def test_xor(self):

diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py
@@ -2652,7 +2652,7 @@ def rsqrt(self):
     print(Tensor([1., 2., 3., 4.]).rsqrt().numpy())
     ```
     """
-    return self.reciprocal().sqrt()
+    return self.sqrt().reciprocal()
   def sin(self):
     """
     Computes the sine of the tensor element-wise.