github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-19 02:44:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

move metal tc check to renderer [pr] (#7248)

Browse Source
This commit is contained in:
George Hotz
2024-10-24 09:38:57 +07:00
committed by GitHub
parent 451c043552
commit 72ddcdb4d1
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_metal.py
View File

@@ -172,8 +172,8 @@ class MetalDevice(Compiled):
self.timeline_value = 0
from tinygrad.runtime.graph.metal import MetalGraph
super().__init__(device, MetalAllocator(self), MetalRenderer(), MetalCompiler(None if getenv("METAL_XCODE") else self),
functools.partial(MetalProgram, self), MetalGraph)
super().__init__(device, MetalAllocator(self), MetalRenderer(os.uname().machine == "arm64"),
MetalCompiler(None if getenv("METAL_XCODE") else self), functools.partial(MetalProgram, self), MetalGraph)
def synchronize(self):
for cbuf in self.mtl_buffers_in_flight: wait_check(cbuf)
self.mv_in_metal.clear()