split grouper and kernelize [pr] (#10854)

This commit is contained in:
George Hotz
2025-06-17 17:54:20 -07:00
committed by GitHub
parent 75503955bf
commit cba6e15937
11 changed files with 481 additions and 478 deletions

View File

@@ -2,7 +2,7 @@ import sys, onnx
from tinygrad import Tensor, fetch, GlobalCounters
from tinygrad.uop.ops import UOp
from tinygrad.frontend.onnx import OnnxRunner
from tinygrad.engine.grouper import get_kernelize_map
from tinygrad.engine.kernelize import get_kernelize_map
from tinygrad.engine.schedule import create_schedule_with_vars
from tinygrad.engine.realize import run_schedule