ROCm don't override user set HSA_OVERRIDE_GFX_VERSION

pull/3982/head
Disty0 2025-06-10 13:21:37 +03:00
parent f5b575db28
commit d2ffee1b4e
2 changed files with 2 additions and 1 deletions

View File

@ -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`

View File

@ -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()