Skip to content

Commit 189bfa1

Browse files
authored
enable backward test for pow(neg const ** x) (tinygrad#8912)
backward works now. 0**x still does not work because it's a special case fixed in transcendental
1 parent 9307572 commit 189bfa1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -607,6 +607,7 @@ def test_pow(self):
607607
helper_test_op([], lambda: b**1.1, lambda: a**1.1)
608608

609609
def test_pow_const(self):
610+
helper_test_op([(45,65)], lambda x: x**0.0)
610611
helper_test_op([(45,65)], lambda x: x**1.0)
611612
helper_test_op([(45,65)], lambda x: x**-1.0)
612613
helper_test_op([(45,65)], lambda x: 1.0**x)
@@ -616,8 +617,7 @@ def test_pow_const(self):
616617
helper_test_op([()], lambda x: 2.0**x)
617618
# TODO: fix backward
618619
helper_test_op(None, lambda x: 0**x, vals=[[-2.,-1,0,1,2,3]], forward_only=True)
619-
# TODO: fix backward, should be nan
620-
helper_test_op(None, lambda x: (-2)**x, vals=[[-2.,-1,0,1,2,3]], forward_only=True)
620+
helper_test_op(None, lambda x: (-2)**x, vals=[[-2.,-1,0,1,2,3]])
621621

622622
@unittest.skip("not supported")
623623
def test_pow_int(self):

0 commit comments

Comments
 (0)