am: download regs (#10419)

* am: download regs

* x

* linter

* mypy

* after merge

* raise

* fixed name

* fix

* xx

* remove

* missing reg

* missing reg

* move to online

* ops
This commit is contained in:
nimlgen
2025-05-20 18:59:56 +03:00
committed by GitHub
parent 965f9e0696
commit 2895198c36
25 changed files with 48 additions and 1417994 deletions

View File

@@ -1,7 +1,6 @@
import unittest
from tinygrad.runtime.support.am.amdev import AMMemoryManager, AMPageTableTraverseContext
from tinygrad.runtime.support.am.ip import AM_GMC
from tinygrad.runtime.support.amd import import_module
from tinygrad.runtime.support.hcq import MMIOInterface
from tinygrad.runtime.autogen.am import am
from tinygrad.helpers import mv_address
@@ -177,19 +176,5 @@ class TestAMPageTable(unittest.TestCase):
must_cover_checker(va, sz)
not_cover_checker(va, sz)
class TestAM(unittest.TestCase):
def test_imports(self):
with self.assertRaises(ImportError): import_module("gc", (7, 0, 0))
x = import_module("gc", (11, 0, 0))
assert x.__name__ == "tinygrad.runtime.autogen.am.gc_11_0_0"
x = import_module("gc", (11, 6, 0))
assert x.__name__ == "tinygrad.runtime.autogen.am.gc_11_0_0"
x = import_module("gc", (12, 0, 0))
assert x.__name__ == "tinygrad.runtime.autogen.am.gc_12_0_0"
x = import_module("gc", (10, 3, 0))
assert x.__name__ == "tinygrad.runtime.autogen.am.gc_10_3_0"
x = import_module("gc", (10, 3, 3))
assert x.__name__ == "tinygrad.runtime.autogen.am.gc_10_3_0"
if __name__ == "__main__":
unittest.main()