Function import_module

tinygrad/runtime/support/amd.py:31–37

(name:str, target:tuple[int, int, int], submod="")

# this is not universally correct, see below for an example, but appears reliable for most recent gpus
# https://github.com/torvalds/linux/blob/9207d47f966be9f4d52e7e0119ac2b7a7e366f3e/drivers/gpu/drm/amd/amdgpu/amdgpu_discovery.c#L3163
def import_module(name:str, target:tuple[int, int, int], submod=""):
  # version overrides
  target = {("smu", (13, 0, 7)): (13, 0, 0)}.get((name, target), target)
  mod = getattr(tinygrad.runtime.autogen.am, submod) if submod else tinygrad.runtime.autogen.am
  if (children:=[c for c in mod.__all__ if c.startswith(name) and (v:=tuple(map(int, c.split('_')[1:])))[0] == target[0] and v <= target]):
    return getattr(mod, children[-1])
  raise ImportError(f"Failed to import {submod+'.' if submod else ''}{name} {'.'.join(map(str, target))}")

def import_soc(ip): return getattr(tinygrad.runtime.autogen.am, f"soc_{ip[0]}")