From e3b3cb163d859981ad24e257d3bed68650ee63eb Mon Sep 17 00:00:00 2001 From: George Hotz Date: Thu, 1 Jan 2026 12:12:47 -0500 Subject: [PATCH] fix emu test --- extra/assembly/amd/test/test_compare_emulators.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extra/assembly/amd/test/test_compare_emulators.py b/extra/assembly/amd/test/test_compare_emulators.py index 6f10c75575..12b05805ee 100644 --- a/extra/assembly/amd/test/test_compare_emulators.py +++ b/extra/assembly/amd/test/test_compare_emulators.py @@ -9,7 +9,7 @@ os.environ["AMD"] = "1" os.environ["MOCKGPU"] = "1" os.environ["PYTHON_REMU"] = "1" -from extra.assembly.amd.emu import WaveState, decode_program, step_wave, WAVE_SIZE, set_valid_mem_ranges +from extra.assembly.amd.emu import WaveState, decode_program, step_wave, WAVE_SIZE, set_valid_mem_ranges, LDSMem from extra.assembly.amd.test.helpers import KernelInfo REMU_PATH = Path(__file__).parents[3] / "remu/target/release/libremu.so" @@ -99,7 +99,7 @@ class PythonEmulator: self.program = decode_program(kernel) self.state = WaveState() self.state.exec_mask = (1 << n_lanes) - 1 - self.lds = bytearray(65536) + self.lds = LDSMem(bytearray(65536)) self.n_lanes = n_lanes def step(self) -> int: