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 assign preload out of schedule item [pr] (#8710)

Browse Source 
* move assign preload out of schedule item [pr]

* fix that
This commit is contained in:
qazal
2025-01-22 05:43:57 -05:00
committed by GitHub
parent 9a9079118e
commit e3d1464ba4
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_schedule.py
View File

@@ -1844,7 +1844,7 @@ def run_tensor_ast(r:Tensor):
sink = UOp(Ops.STORE, src=(glbl, ShapeTracker.from_shape(r.lazydata.base.shape).to_uop(), r.lazydata.base)).sink()
sink = graph_rewrite(sink, remove_movement_ops+ops_folding+load_buffers+view_left, bufs:=[output])
sink = graph_rewrite(sink, remove_movement_ops+ops_folding+view_right)
si = ScheduleItem(sink, tuple(x.buffer for x in bufs), (), ())
si = ScheduleItem(sink, tuple(x.buffer for x in bufs), ())
run_schedule([si])
return output.realized.as_buffer().cast(output.dtype.fmt, r.shape).tolist()