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

hotfix: don't include the old input_rawbuffers in all_resources

Browse Source
This commit is contained in:
George Hotz
2024-08-01 09:00:11 -07:00
parent d609206a4a
commit 0e34d83777
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/graph/metal.py
View File

@@ -35,7 +35,7 @@ class MetalGraph(GraphRunner):
icb_command.setComputePipelineState_(unwrap2(
self.device.device.newComputePipelineStateWithDescriptor_options_reflection_error_(descriptor, Metal.MTLPipelineOption(0), None, None)))
for i,b in enumerate(ji.bufs):
if b is not None:
if b is not None and b not in input_rawbuffers:
icb_command.setKernelBuffer_offset_atIndex_(b._buf.buf, b._buf.offset, i)
all_resources.append(b._buf.buf)
for i,v in enumerate(prg.p.vars): icb_command.setKernelBuffer_offset_atIndex_(self.int_buf.buf, self.vars.index(v)*4, len(ji.bufs)+i)