mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-11 23:25:04 -05:00
@@ -16,7 +16,7 @@ from extra.onnx import get_run_onnx
|
||||
from tinygrad import Tensor, Device, GlobalCounters, dtypes
|
||||
from tinygrad.dtype import ImageDType
|
||||
from tinygrad.helpers import partition, Context, fetch, getenv, GRAPH, DEBUG
|
||||
from tinygrad.realize import run_schedule, lower_schedule_item, create_schedule
|
||||
from tinygrad.engine.realize import run_schedule, lower_schedule_item, create_schedule
|
||||
from tinygrad.ops import LoadOps, ScheduleItem
|
||||
Device.DEFAULT = "GPU"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user