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 "ops_ext to replace cpu import (#3406)" (#3408)

Browse Source 
This reverts commit 91eb93f85a.
This commit is contained in:
George Hotz
2024-02-15 12:16:10 +01:00
committed by GitHub
parent 91eb93f85a
commit 6356474d6d
9 changed files with 11 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/abstractions.py
View File

@@ -135,8 +135,8 @@ assert len(lazyop.srcs) == 2
# the source is a LazyBuffer that is a "CPU" Tensor
# again, a LazyOp AST is like a GPU kernel. you have to copy the data on the device first
assert lazyop.srcs[0].op == LoadOps.COPY
assert lazyop.srcs[0].srcs[0].device == "EXT"
assert lazyop.srcs[0].srcs[0].realized._buf[0][0] == 2, "the src of the COPY LazyOP is a LazyBuffer on the CPU holding [2]"
assert lazyop.srcs[0].srcs[0].device == "CPU"
assert lazyop.srcs[0].srcs[0].realized._buf[0] == 2, "the src of the COPY LazyOP is a LazyBuffer on the CPU holding [2]"
assert result.lazydata.base.realized is None, "the LazyBuffer is not realized yet"
# now we realize the LazyBuffer