9c7833faf9
* Vegetarian Filet o Fish model * fix.. atc.. * test cluster_speed_limit * fix.. cluster_speed_limit.. 2 * fix.. clusterspeedlimit3 * cruise speed to roadlimit speed * fix.. * fix.. eng * deltaUp/Down for lanechange * fix.. atc desire... * fix.. * ff * ff * fix.. * fix.. eng * fix engsound * Update desire_helper.py * fix.. connect... * fix curve_min speed * Revert "fix curve_min speed" This reverts commit fcc9c2eb14eb3504abef3e420db93e8882e56f37. * Reapply "fix curve_min speed" This reverts commit 2d2bba476c58a7b4e13bac3c3ad0e4694c95515d. * fix.. auto speed up.. roadlimit * fix.. atc auto lanechange... * Update desire_helper.py * Update cruise.py * debug atc... * fix.. waze alert offset.. * fix.. * test atc.. * fix.. * fix.. atc * atc test.. * fix.. atc * fix.. atc2 * fix.. atc3 * KerryGold Model. latsmooth_sec = 0.0 * lat smooth seconds 0.13 * fix comment * fix.. auto cruise, and speed unit * change lanemode switching. * erase mazda lkas button.
29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
import unittest
|
|
from tinygrad import Device, Tensor
|
|
from tinygrad.engine.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_buffer().cast("f")[0]
|
|
assert val == 4000.0, f"got val {val}"
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|
|
|