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

rsqrt is self.reciprocal().sqrt() (#2790)

Browse Source 
(1/self) is incorrect for int tensor
This commit is contained in:
chenyu
2023-12-16 01:58:05 -05:00
committed by GitHub
parent c5fa9eb36e
commit bb6f7b6172
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_dtype.py
View File

@@ -312,14 +312,14 @@ class TestAutoCastType(unittest.TestCase):
def test_int_to_float_unary_func(self, dtype):
for func in [
lambda t: t.exp(),
# lambda t: t.exp2(), # requires MUL
lambda t: t.exp2(),
lambda t: t.log(),
lambda t: t.log2(),
lambda t: t.sqrt(),
# lambda t: t.rsqrt(), # requires DIV
lambda t: t.rsqrt(),
lambda t: t.sin(),
# lambda t: t.cos(), # requires SUB
# lambda t: t.tan(), # requires .cos() to work
lambda t: t.cos(),
lambda t: t.tan(),
lambda t: t.sigmoid(),
]:
a = [2, 3, 4]