github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 23:18:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

amd: use getattr for autogen (#13630)

Browse Source 
* amd: use getattr for autogen

* fi
This commit is contained in:
nimlgen
2025-12-09 20:36:26 +03:00
committed by GitHub
parent 76d465dbc3
commit ddecba300f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/support/amd.py
View File

@@ -1,4 +1,4 @@
import functools, importlib, re, urllib
import functools, re, urllib, tinygrad.runtime.autogen
from collections import defaultdict
from dataclasses import dataclass
from tinygrad.helpers import getbits, fetch
@@ -53,7 +53,7 @@ def import_header(path:str, url=AMDGPU_URL):
def import_module(name:str, version:tuple[int, ...], version_prefix:str=""):
for ver in fixup_ip_version(name, version):
try: return importlib.import_module(f"tinygrad.runtime.autogen.am.{name}_{version_prefix}{'_'.join(map(str, ver))}")
try: return getattr(tinygrad.runtime.autogen.am, f"{name}_{version_prefix}{'_'.join(map(str, ver))}")
except ImportError: pass
raise ImportError(f"Failed to load autogen module for {name.upper()} {'.'.join(map(str, version))}")