mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix division by zero in add views (#11075)

Browse source
This commit is contained in:
Sieds Lykles 2025-07-03 20:37:59 +02:00 • committed by GitHub
commit 19f07e72f6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tinygrad/shape/view.py
View file

@ -198,7 +198,7 @@ class View:
for d1, st in enumerate(vm1.strides):
if st == 0: continue
for d2, (o, s1) in enumerate(zip(origin, unravel(vm2.shape, vm1.offset + st))):
if (s1 := s1 - o) == 0: continue
if not resolve((s1 := s1 - o)!=0): continue # if s1 can possible be 0
terms[d2].append((d1, s1))
strides[d1] += s1 * vm2.strides[d2]