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

mask out writable bufs in runtime access_resources (#7234)

Browse Source
This commit is contained in:
qazal
2024-10-23 16:13:50 +03:00
committed by GitHub
parent d2b608233a
commit aeeb917b6e
6 changed files with 15 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_graph.py
View File

@@ -54,8 +54,7 @@ def helper_test_graphs(graph_impl, graphs, runs=RUN_CNT):
out_buffers = set()
for graph in graphs:
for ji in graph:
writable_buffers = ji.prg.p.outcount if isinstance(ji.prg, CompiledRunner) else 1
out_buffers.update(ji.bufs[:writable_buffers])
out_buffers.update([ji.bufs[i] for i in (ji.prg.p.outs if isinstance(ji.prg, CompiledRunner) else [0])])
bufs += ji.bufs
reg_ji.append(ji)
bufs = dedup(bufs)