github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 06:58:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

support int Tensor pow to const non-negative int (#11624)

Browse Source 
matches torch
This commit is contained in:
chenyu
2025-08-11 16:50:19 -07:00
committed by GitHub
parent 857a830dcc
commit d623f6d850
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/test_ops.py
View File

@@ -699,6 +699,13 @@ class TestOps(unittest.TestCase):
helper_test_op(None, lambda x: x**-0.3, vals=[[0.0]])
helper_test_op(None, lambda x: x**-1.0, vals=[[-1.0, 0.0, 1.0]])
def test_int_pow_const_int(self):
helper_test_op(None, lambda x: x**0, vals=[[-2,0,2]], forward_only=True)
helper_test_op(None, lambda x: x**1, vals=[[-2,0,2]], forward_only=True)
helper_test_op(None, lambda x: x**2, vals=[[-2,0,2]], forward_only=True)
helper_test_op(None, lambda x: x**29, vals=[[-2,0,2]], forward_only=True)
self.helper_test_exception(None, lambda x: x**-2, vals=[[-2,0,2]], forward_only=True, expected=RuntimeError)
@unittest.skip("not supported")
def test_pow_int(self):
def _test(base, exponent): helper_test_op(None, lambda x,y: x**y, vals=[base, exponent], forward_only=True)