IQ.Pilot Release Commit @ bec7652

artifacts/package_sources/tinygrad/test/external/external_test_amd.py

@@ -0,0 +1,28 @@
+import unittest
+from tinygrad import Device, Tensor
+from tinygrad.schedule import create_schedule
+from tinygrad.runtime.ops_amd import AMDDevice
+
+class TestAMD(unittest.TestCase):
+  @classmethod
+  def setUpClass(self):
+    TestAMD.d0: AMDDevice = Device["AMD"]
+    TestAMD.a = Tensor([0.,1.], device="AMD").realize()
+    TestAMD.b = self.a + 1
+    si = create_schedule([self.b.uop])[-1]
+    TestAMD.d0_runner = TestAMD.d0.get_runner(*si.ast)
+    TestAMD.b.uop.buffer.allocate()
+
+  def test_amd_ring_64bit_doorbell(self):
+    TestAMD.d0.pm4_write_pointer[0] = TestAMD.d0.pm4_write_pointer[0] + (2 << 32) - TestAMD.d0.pm4_ring.size // 4
+    for _ in range(2000):
+      TestAMD.d0_runner.clprg(TestAMD.b.uop.buffer._buf, TestAMD.a.uop.buffer._buf,
+                              global_size=TestAMD.d0_runner.global_size, local_size=TestAMD.d0_runner.local_size)
+      TestAMD.d0_runner.clprg(TestAMD.a.uop.buffer._buf, TestAMD.b.uop.buffer._buf,
+                              global_size=TestAMD.d0_runner.global_size, local_size=TestAMD.d0_runner.local_size)
+    val = TestAMD.a.uop.buffer.as_memoryview().cast("f")[0]
+    assert val == 4000.0, f"got val {val}"
+
+if __name__ == "__main__":
+  unittest.main()
+