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

Revert "bug in metal: offset is stored as uint32, overflow (#15129)" (#15136)

Browse Source 
This reverts commit 9c58db16fa.
This commit is contained in:
chenyu
2026-03-04 16:54:42 -05:00
committed by GitHub
parent 9c58db16fa
commit 34594bcaaf
2 changed files with 1 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
test/unit/test_metal_graph.py
View File

@@ -1,26 +0,0 @@
import unittest
from unittest.mock import MagicMock
from tinygrad import Device
from tinygrad.engine.realize import CompiledRunner
def _ei(*offsets):
ei = MagicMock()
ei.prg = MagicMock(spec=CompiledRunner)
ei.bufs = [None if o is None else MagicMock(**{"_buf.offset": o}) for o in offsets]
return ei
@unittest.skipUnless(Device.DEFAULT == "METAL", "Metal device required to run")
class TestMetalGraph(unittest.TestCase):
def setUp(self):
from tinygrad.runtime.graph.metal import MetalGraph
self.MetalGraph = MetalGraph
self.dev = Device[Device.DEFAULT]
def test_supports_exec_item_normal_offset(self):
assert self.MetalGraph.supports_exec_item([self.dev], _ei(0, 100, 0xFFFFFFFF)) is True
def test_supports_exec_item_overflow_offset(self):
assert self.MetalGraph.supports_exec_item([], _ei(0, 0x100000000)) is False
if __name__ == "__main__":
unittest.main()