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

Revert "hotfix: remove setrecursionlimit"

Browse Source 
This reverts commit acfc81642a.
This commit is contained in:
George Hotz
2025-06-16 16:01:42 -07:00
parent acfc81642a
commit b5ce227850
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/engine/grouper.py
View File

@@ -11,6 +11,10 @@ from tinygrad.engine.multi import multi_pm
from tinygrad.shape.shapetracker import ShapeTracker
from tinygrad.shape.view import View, strides_for_shape
# creation can recurse a lot
import sys
sys.setrecursionlimit(10000)
# **** schedule simplifier
def simplify_stride0_reduce(reduce:UOp, x:UOp):