github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

default threefry (#6116)

Browse Source
This commit is contained in:
wozeparrot
2024-09-25 17:45:13 +08:00
committed by GitHub
parent 992cde05d7
commit c100f3d406
15 changed files with 82 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_arange.py
View File

@@ -142,8 +142,8 @@ class TestIndexing(unittest.TestCase):
from tinygrad.nn.datasets import mnist
X_train, Y_train, _, _ = mnist()
with Context(NOOPT=noopt, FUSE_ARANGE=1, SPLIT_REDUCEOP=0):
samples = Tensor.randint(getenv("BS", 512), high=X_train.shape[0]).realize()
GlobalCounters.reset()
samples = Tensor.randint(getenv("BS", 512), high=X_train.shape[0])
x = X_train[samples].numpy()
y = Y_train[samples].numpy()
assert GlobalCounters.global_ops < op_limit, f"too many ops {GlobalCounters.global_ops} != {op_limit}"