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

Move rawbuffer (#697)

Browse Source 
* move GlobalCounters to helpers

* that's not part of the public api

* move InterpretedBuffer

* remove fromCPU from devicebuffer
This commit is contained in:
George Hotz
2023-03-13 22:30:36 -07:00
committed by GitHub
parent cbc5a7222a
commit 54f499b623
17 changed files with 162 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/unit/test_graph.py
View File

@@ -14,8 +14,8 @@ class TestGraph(unittest.TestCase):
assert nx.is_isomorphic(G, RG, node_match=lambda x,y: x["label"] == y["label"], edge_match=lambda x,y: x["label"] == y["label"] if "label" in y else True)
def test_add_graph(self):
a = CPUBuffer.fromCPU(np.ones((4,4), dtype=np.float32))
b = CPUBuffer.fromCPU(np.ones((4,4), dtype=np.float32))
a = CPUBuffer(np.ones((4,4), dtype=np.float32))
b = CPUBuffer(np.ones((4,4), dtype=np.float32))
ast = LazyOp(BinaryOps.ADD, (a,b))
ret = CPUBuffer(np.ones((4,4), dtype=np.float32))
@@ -30,8 +30,8 @@ class TestGraph(unittest.TestCase):
self.helper_compare_graph(RG)
def test_add_sum_graph(self):
a = CPUBuffer.fromCPU(np.ones((4,4), dtype=np.float32))
b = CPUBuffer.fromCPU(np.ones((1,1), dtype=np.float32))
a = CPUBuffer(np.ones((4,4), dtype=np.float32))
b = CPUBuffer(np.ones((1,1), dtype=np.float32))
op0 = LazyOp(MovementOps.RESHAPE, (b,), (4, 4))
op1 = LazyOp(BinaryOps.ADD, (a,op0))
ast = LazyOp(ReduceOps.SUM, (op1,), (1,1))
@@ -48,12 +48,12 @@ class TestGraph(unittest.TestCase):
self.helper_compare_graph(RG)
def test_add_graph_prune(self):
a = CPUBuffer.fromCPU(np.ones((1,1), dtype=np.float32))
a = CPUBuffer(np.ones((1,1), dtype=np.float32))
ast = LazyOp(MovementOps.RESHAPE, (a,), (4, 4))
ret = CPUBuffer(np.ones((4,4), dtype=np.float32))
log_op(ret, ast, show_graph=True)
b = CPUBuffer.fromCPU(np.ones((4,4), dtype=np.float32))
b = CPUBuffer(np.ones((4,4), dtype=np.float32))
ast = LazyOp(BinaryOps.ADD, (ret,b))
ret = CPUBuffer(np.ones((4,4), dtype=np.float32))
log_op(ret, ast, show_graph=True)