ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Simplify MI300X startup banner name (#4049)

Browse source 
* Improve HIP GPU name reporting in startup banner

* Drop MI300X arch suffix in banner name

* Normalize _utils.py file mode

* Simplify FA2 fallback text and filter AMD ids noise

* Strip trailing GPU arch suffix via regex

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Use gfx lookup default and normalize Ryzen AI naming

* Remove name-path Ryzen AI normalization

* Expand ROCm gfx map to full documented GPU name aliases

* Simplify HIP fallback naming to AMD gfx token

* Remove Ryzen Al torch_name normalization

---------

Co-authored-by: Daniel Hanchen <danielhanchen@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Daniel Han 2026-02-14 02:24:03 -08:00 • committed by GitHub
commit 596683e7ca
2 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/import_fixes.py
View file

@ -199,6 +199,8 @@ if os.environ.get("UNSLOTH_ENABLE_LOGGING", "0") != "1":
# Triton "df: No such file or directory" stderr noise
sys.stderr.add_filter("df: No such file")
# ROCm/libdrm missing ids table stderr noise on some AMD setups
sys.stderr.add_filter("amdgpu.ids: No such file or directory")
# Fix up AttributeError: 'MessageFactory' object has no attribute 'GetPrototype'

24
unsloth/models/_utils.py
View file

@ -154,12 +154,14 @@ from unsloth_zoo.training_utils import (
def resolve_hip_gpu_stats_name(gpu_stats):
name = str(getattr(gpu_stats, "name", "") or "").strip()
name = re.sub(r"\s*\([^)]*\)\s*$", "", name).strip()
normalized_name = name.lower().strip(". ")
if normalized_name and normalized_name not in ("amd radeon graphics",):
return name + ". "
try:
torch_name = str(torch.cuda.get_device_name(0) or "").strip()
torch_name = re.sub(r"\s*\([^)]*\)\s*$", "", torch_name).strip()
except Exception:
torch_name = ""
normalized_torch_name = torch_name.lower().strip(". ")
@ -174,11 +176,11 @@ def resolve_hip_gpu_stats_name(gpu_stats):
break
if arch_name:
# gfx942 maps to MI300X on current ROCm naming.
if arch_name.lower().startswith("gfx942"):
return f"AMD Instinct MI300X ({arch_name}). "
return f"AMD GPU ({arch_name}). "
return "AMD GPU Device. "
arch_name = arch_name.strip()
match = re.search(r"(gfx[0-9a-z]+)", arch_name, flags = re.I)
if match:
return f"AMD {match.group(1).lower()} GPU. "
return "AMD GPU. "
from unsloth_zoo.temporary_patches import (
@ -874,10 +876,8 @@ if DEVICE_TYPE == "cuda":
)
except:
print(
"Unsloth: Your Flash Attention 2 installation seems to be broken?\n"
"A possible explanation is you have a new CUDA version which isn't\n"
"yet compatible with FA2? Please file a ticket to Unsloth or FA2.\n"
"We shall now use Xformers instead, which does not have any performance hits!"
"Unsloth: Your Flash Attention 2 installation seems to be broken. "
"Using Xformers instead. No performance changes will be seen."
)
# Stop Flash Attention from importing!
@ -925,10 +925,8 @@ elif DEVICE_TYPE == "hip":
)
except:
print(
"Unsloth: Your Flash Attention 2 installation seems to be broken?\n"
"A possible explanation is you have a new CUDA version which isn't\n"
"yet compatible with FA2? Please file a ticket to Unsloth or FA2.\n"
"We shall now use Xformers instead, which does not have any performance hits!"
"Unsloth: Your Flash Attention 2 installation seems to be broken. "
"Using Xformers instead. No performance changes will be seen."
)
# Stop Flash Attention from importing!