github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

move hcq device to runtime [pr] (#6879)

Browse Source 
* things that are only used in one place don't belong in helpers [pr]

* start moving hcq device [pr]

* fix paths
This commit is contained in:
George Hotz
2024-10-04 22:26:50 +08:00
committed by GitHub
parent 5be2bd18a6
commit 6b063450df
9 changed files with 563 additions and 554 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_hcq.py
View File

@@ -1,7 +1,8 @@
import unittest, ctypes, struct
from tinygrad import Device, Tensor, dtypes
from tinygrad.helpers import CI, getenv
from tinygrad.device import Buffer, BufferOptions, HCQCompiled
from tinygrad.device import Buffer, BufferOptions
from tinygrad.runtime.support.hcq import HCQCompiled
from tinygrad.engine.schedule import create_schedule
from tinygrad.engine.realize import get_runner, CompiledRunner
from tinygrad.codegen.kernel import Kernel, Opt, OptOps

3
test/test_profiler.py
View File

@@ -1,7 +1,8 @@
import unittest, struct, contextlib, tempfile, pathlib, json, time, atexit, random
from tinygrad import Device, Tensor, dtypes, TinyJit
from tinygrad.helpers import CI, getenv, Context
from tinygrad.device import Buffer, BufferOptions, ProfileLogger, HCQCompiled
from tinygrad.device import Buffer, BufferOptions
from tinygrad.runtime.support.hcq import ProfileLogger, HCQCompiled
from tinygrad.engine.schedule import create_schedule
from tinygrad.engine.realize import get_runner