github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

rename to extra.assembly.amd (#13879)

Browse Source
This commit is contained in:
George Hotz
2025-12-29 14:10:55 -05:00
committed by GitHub
parent 37f0fa11b6
commit 81cf9ea0ab
40 changed files with 5167 additions and 795 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/mockgpu/helpers.py
View File

@@ -21,7 +21,7 @@ class PythonRemu:
rsrc2: int = 0x19c # Default: USER_SGPR_COUNT=14, enable X and Y workgroup IDs
def run_asm(self, lib: int, lib_sz: int, gx: int, gy: int, gz: int, lx: int, ly: int, lz: int, args_ptr: int) -> int:
from extra.assembly.rdna3.emu import run_asm, set_valid_mem_ranges
from extra.assembly.amd.emu import run_asm, set_valid_mem_ranges
# Pad ranges to handle GPU loads that may read past small buffers (e.g. s_load_b128 on 12-byte buffer)
set_valid_mem_ranges({(start, size + 4096) for start, size in self.valid_mem_ranges})
return run_asm(lib, lib_sz, gx, gy, gz, lx, ly, lz, args_ptr, self.rsrc2)

2
test/testextra/test_cfg_viz.py
View File

@@ -10,7 +10,7 @@ from tinygrad.renderer import ProgramSpec
from tinygrad.helpers import TracingKey, getenv
from tinygrad.engine.realize import ExecItem, CompiledRunner
from extra.assembly.rdna3.autogen import *
from extra.assembly.amd.autogen.rdna3 import *
# TODO: use the RDNA3 renderer when it's in master
template = """.text