diff --git a/CHANGELOG.md b/CHANGELOG.md index e4cd04bcf..7ded29486 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,7 @@ - Increase the medvram mode threshold from 8GB to 12GB - Set CPU backend to use FP32 by default - Relax Python version checks for Zluda + - don't override user set gfx version with ROCm - **Torch** - set default to `torch==2.7.1` diff --git a/installer.py b/installer.py index f92082dbe..5eb1a86f3 100644 --- a/installer.py +++ b/installer.py @@ -696,7 +696,7 @@ def install_rocm_zluda(): log.debug(f'ROCm hipBLASLt: arch={device.name} available={device.blaslt_supported}') rocm.set_blaslt_enabled(device.blaslt_supported) - if device is None: + if device is None or os.environ.get("HSA_OVERRIDE_GFX_VERSION", None) is not None: log.debug('ROCm: HSA_OVERRIDE_GFX_VERSION auto config skipped') else: gfx_ver = device.get_gfx_version()