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

remove real_offset (#2234)

Browse Source 
* remove real_offset

* pass in numnode

* remove that real_offset

* sample only for variable
This commit is contained in:
George Hotz
2023-11-07 17:30:53 -08:00
committed by GitHub
parent 4d95e6d070
commit c0a033f01d
7 changed files with 36 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/external/fuzz_shapetracker.py vendored
View File

@@ -1,5 +1,5 @@
import random
from tinygrad.helpers import DEBUG
from tinygrad.helpers import DEBUG, getenv
from test.unit.test_shapetracker import CheckingShapeTracker
random.seed(42)
@@ -55,7 +55,7 @@ def do_expand(st):
if __name__ == "__main__":
ops = [do_permute, do_pad, do_shrink, do_reshape_split_one, do_reshape_combine_two, do_stride, do_expand]
for _ in range(200):
for _ in range(getenv("CNT", 200)):
st = CheckingShapeTracker((random.randint(2, 10), random.randint(2, 10), random.randint(2, 10)))
for i in range(8): random.choice(ops)(st)
st.assert_same()