github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 22:08:08 -05:00
Code Issues Packages Projects Releases 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
parent 678cabc6f2
commit 19f07e72f6
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]