1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
"""Enumerate every IP block and its base addresses from the IP discovery table.
Purpose: answer "what other mailboxes exist on SMN" with the card's own
authoritative list instead of guessing address neighbourhoods.
SAFETY: deliberately does NOT hand-read BAR0 at (vram_size - 64KiB). On the
healthy card vram_size is 0x5ff000000, so that offset is ~24 GB -- far outside
the BAR0 aperture, and a high-offset BAR0 read raises a TLP error that wedges
PCIe until a power cycle. Instead we use tinygrad's normal AMDev path, which
reads the table through the supported windowed accessor and parses it.
READ-ONLY with respect to our own pokes; device init is tinygrad's standard path.
"""
import sys
sys.path.insert(0, "/Users/user/src/tinygrad")
from tinygrad import Device
from tinygrad.runtime.autogen.am import am
d = Device["AMD"]
impl = None
for attr in ("dev_iface", "iface"):
o = getattr(d, attr, None)
if o is None: continue
impl = getattr(o, "dev_impl", None) or o
if hasattr(impl, "regs_offset"): break
if impl is None or not hasattr(impl, "regs_offset"):
sys.exit("could not reach AMDev.regs_offset (got %r)" % (impl,))
print("vram_size = %#x (%d MB)" % (impl.vram_size, impl.vram_size >> 20))
print()
name = {}
for k in dir(am):
if k.endswith("_HWIP"):
try: name[getattr(am, k)] = k[:-5]
except Exception: pass
print("%-14s %-12s %-8s %s" % ("IP", "version", "inst", "base addresses"))
rows = []
for hw_ip, insts in sorted(impl.regs_offset.items()):
ver = impl.ip_ver.get(hw_ip)
for inst, bases in sorted(insts.items()):
rows.append((name.get(hw_ip, str(hw_ip)), ver, inst, bases))
print("%-14s %-12s %-8s %s"
% (name.get(hw_ip, str(hw_ip)),
".".join(map(str, ver)) if ver else "?",
inst,
" ".join("%#x" % b for b in bases)))
print("\n=== distinct base addresses (candidate SMN targets) ===")
allb = sorted({b for _, _, _, bases in rows for b in bases})
print("count =", len(allb))
for b in allb: print(" %#x" % b)
|
