extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-08-23 18:05:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Allow a user to temporarily override the HSA_OVERRIDE_GFX_VERSION and HIP_VISIBLE_DEVICES env variables, until ED gets proper per-process multi-gpu support

Browse Source
This commit is contained in:
cmdr2
2025-08-22 14:13:16 +05:30
parent a26e96eafc
commit b5b68670b3
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/check_modules.py
View File

@@ -302,6 +302,16 @@ def launch_uvicorn():
if hasattr(torchruntime, "info"):
torchruntime.info()
# allow a user to override the HSA_OVERRIDE_GFX_VERSION and HIP_VISIBLE_DEVICES variables
# until ED gets process-based multi-GPU support (which will allow different processes to use different GPUs)
backend_config = config.get("backend_config", {})
if "HSA_OVERRIDE_GFX_VERSION" in backend_config:
os.environ["HSA_OVERRIDE_GFX_VERSION"] = str(backend_config["HSA_OVERRIDE_GFX_VERSION"])
print(f"backend_config overrode HSA_OVERRIDE_GFX_VERSION to {os.environ['HSA_OVERRIDE_GFX_VERSION']}")
if "HIP_VISIBLE_DEVICES" in backend_config:
os.environ["HIP_VISIBLE_DEVICES"] = str(backend_config["HIP_VISIBLE_DEVICES"])
print(f"backend_config overrode HIP_VISIBLE_DEVICES to {os.environ['HIP_VISIBLE_DEVICES']}")
if os_name == "Windows":
os.environ["PYTHONPATH"] = str(Path(os.environ["INSTALL_ENV_DIR"], "lib", "site-packages"))
else: