mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-07 21:26:21 -05:00
8b7ecd63bb
* no zeroview start * closer * stride mask * st tests pass, delete ZeroView * byebye zv * close to working * not contiguous with mask * subtract, don't add * mask on view * ugh, that shouldn't have been in there * shape merge * bugfixes * fuzzer + 4 fuzzer failures * fuzzer for symbolic * more fuzzing and nothing * that fuzzer doesn't hit either * fixes padding...ugh * no more offsets * working * rewrite load and store * all checks * fix idxs * progress * bugfix * float4_axis * works * cleanups * complex valids_okay
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
import random
|
|
from test.unit.test_shapetracker import CheckingShapeTracker
|
|
|
|
def do_permute(st):
|
|
perm = list(range(0, len(st.shape)))
|
|
random.shuffle(perm)
|
|
perm = tuple(perm)
|
|
print("st.permute(", perm, ")")
|
|
st.permute(perm)
|
|
|
|
def do_pad(st):
|
|
c = random.randint(0, len(st.shape)-1)
|
|
pad = tuple((random.randint(0,2), random.randint(0,2)) if i==c else (0,0) for i in range(len(st.shape)))
|
|
print("st.pad(", pad, ")")
|
|
st.pad(pad)
|
|
|
|
def do_reshape_split_one(st):
|
|
c = random.randint(0, len(st.shape)-1)
|
|
poss = [n for n in [1,2,3,4,5] if st.shape[c]%n == 0]
|
|
spl = random.choice(poss)
|
|
shp = st.shape[0:c] + (st.shape[c]//spl, spl) + st.shape[c+1:]
|
|
print("st.reshape(", shp, ")")
|
|
st.reshape(shp)
|
|
|
|
def do_reshape_combine_two(st):
|
|
if len(st.shape) < 2: return
|
|
c = random.randint(0, len(st.shape)-2)
|
|
shp = st.shape[:c] + (st.shape[c] * st.shape[c+1], ) + st.shape[c+2:]
|
|
print("st.reshape(", shp, ")")
|
|
st.reshape(shp)
|
|
|
|
def do_shrink(st):
|
|
c = random.randint(0, len(st.shape)-1)
|
|
while 1:
|
|
shrink = tuple((random.randint(0,s), random.randint(0,s)) if i == c else (0,s) for i,s in enumerate(st.shape))
|
|
if all(x<y for (x,y) in shrink): break
|
|
print("st.shrink(", shrink, ")")
|
|
st.shrink(shrink)
|
|
|
|
def do_stride(st):
|
|
c = random.randint(0, len(st.shape)-1)
|
|
stride = tuple(random.choice([-2,-1,2]) if i==c else 1 for i in range(len(st.shape)))
|
|
print("st.stride(", stride, ")")
|
|
st.stride(stride)
|
|
|
|
def do_expand(st):
|
|
c = [i for i,s in enumerate(st.shape) if s==1]
|
|
if len(c) == 0: return
|
|
c = random.choice(c)
|
|
expand = tuple(random.choice([2,3,4]) if i==c else s for i,s in enumerate(st.shape))
|
|
print("st.expand(", expand, ")")
|
|
st.expand(expand)
|
|
|
|
if __name__ == "__main__":
|
|
ops = [do_permute, do_pad, do_shrink, do_reshape_split_one, do_reshape_combine_two, do_stride, do_expand]
|
|
while 1:
|
|
st = CheckingShapeTracker((3, 3, 3))
|
|
for i in range(8): random.choice(ops)(st)
|
|
#st.simplify()
|
|
st.assert_same()
|