github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

add failed Tensor.pow test cases (#3334)

Browse Source 
tried refactoring pow and found some bugs
This commit is contained in:
chenyu
2024-02-07 04:28:24 -05:00
committed by GitHub
parent ea74856d99
commit ca66be6a70
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_ops.py
View File

@@ -383,6 +383,10 @@ class TestOps(unittest.TestCase):
helper_test_op([(45,65)], lambda x: 2.0**x)
helper_test_op([()], lambda x: x**2.0)
helper_test_op([()], lambda x: 2.0**x)
# TODO: fix 0**x and 0**0 == 1
# helper_test_op(None, lambda x: 0**x, vals=[[-2.,-1,0,1,2,3]])
# TODO: fix backward, should be nan
helper_test_op(None, lambda x: (-2)**x, vals=[[-2.,-1,0,1,2,3]], forward_only=True)
def test_sqrt(self):
helper_test_op([(45,65)], lambda x: x.sqrt())