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

bring buffer back to device (#4517)

Browse Source
This commit is contained in:
George Hotz
2024-05-10 11:22:31 -07:00
committed by GitHub
parent a2b707a3eb
commit d438d5698d
26 changed files with 157 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/external_test_hcq.py vendored
View File

@@ -1,7 +1,7 @@
import unittest, ctypes, struct, time, array
from tinygrad import Device, Tensor, dtypes
from tinygrad.helpers import to_mv
from tinygrad.buffer import Buffer, BufferOptions
from tinygrad.device import Buffer, BufferOptions
from tinygrad.engine.schedule import create_schedule
def _time_queue(q, d):

2
test/external/fuzz_schedule.py vendored
View File

@@ -1,7 +1,7 @@
import itertools
import numpy as np
from typing import DefaultDict, Dict, List, Set, Tuple, TypeVar
from tinygrad.buffer import Buffer
from tinygrad.device import Buffer
from tinygrad.engine.realize import CustomOp, capturing, lower_schedule_item
from tinygrad.helpers import DEBUG, MULTIOUTPUT, colored, getenv
from tinygrad.lazy import LazyBuffer

2
test/test_subbuffer.py
View File

@@ -1,7 +1,7 @@
import unittest
from tinygrad import Device, dtypes, Tensor
from tinygrad.helpers import CI
from tinygrad.buffer import Buffer
from tinygrad.device import Buffer
from tinygrad.lazy import view_supported_devices
@unittest.skipIf(Device.DEFAULT not in view_supported_devices, "subbuffer not supported")