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

stop blowing up floats

Browse Source
This commit is contained in:
George Hotz
2022-10-30 16:47:16 -07:00
parent 8afc643bb1
commit db2da22a04
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_ops.py
View File

@@ -6,7 +6,7 @@ import unittest
from tinygrad.tensor import Tensor, Device
FORWARD_ONLY = bool(int(os.getenv("FORWARD_ONLY", "0")))
def helper_test_op(shps, torch_fxn, tinygrad_fxn, atol=1e-6, rtol=1e-3, grad_atol=1e-4, grad_rtol=1e-3, forward_only=False, vals=None, a=-0.5, b=20):
def helper_test_op(shps, torch_fxn, tinygrad_fxn, atol=1e-6, rtol=1e-3, grad_atol=1e-4, grad_rtol=1e-3, forward_only=False, vals=None, a=-0.5, b=3):
torch.manual_seed(0)
np.random.seed(0)
if shps is None: