From f09a4240a1ab24312b3cf1191022d5ca2be75305 Mon Sep 17 00:00:00 2001
From: LeoBorcherding <borchborchmail@gmail.com>
Date: Wed, 6 May 2026 20:43:18 -0500
Subject: [PATCH] fix: register ROCm DLL directory before torch import on
 Windows

Python 3.8+ ignores PATH for extension DLL loading on Windows; amdhip64.dll
and other HIP runtime DLLs must be registered via os.add_dll_directory().
Without this, torch.cuda.is_available() always returns False on AMD ROCm
Windows even when HIP_PATH is correctly set in system environment variables.

Reads HIP_PATH / ROCM_PATH env vars first, then falls back to scanning
common ROCm install roots (C:\Program Files\AMD\ROCm, F:\ROCm, C:\ROCm).
---
 studio/backend/main.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/studio/backend/main.py b/studio/backend/main.py
index cd901327db..0979f48865 100644
--- a/studio/backend/main.py
+++ b/studio/backend/main.py
@@ -12,6 +12,40 @@ from pathlib import Path as _Path
 # Suppress annoying C-level dependency warnings globally
 os.environ["PYTHONWARNINGS"] = "ignore"
 
+# ── Windows AMD ROCm DLL injection ──────────────────────────────────────────
+# On Windows, Python 3.8+ uses a secure DLL search that ignores PATH for
+# extension modules. torch's HIP backend (amdhip64.dll etc.) won't be found
+# even if F:\ROCm\...\bin is in PATH unless we explicitly register the
+# directory with os.add_dll_directory(). Do this before any torch import.
+if sys.platform == "win32":
+    import ctypes as _ctypes
+
+    def _add_rocm_dll_dirs() -> None:
+        hip_path = os.environ.get("HIP_PATH") or os.environ.get("ROCM_PATH")
+        candidates = []
+        if hip_path:
+            candidates.append(os.path.join(hip_path, "bin"))
+        # Also scan common install roots in case HIP_PATH is not set
+        for _root in (r"C:\Program Files\AMD\ROCm", r"F:\ROCm", r"C:\ROCm"):
+            try:
+                if os.path.isdir(_root):
+                    for _ver in sorted(os.listdir(_root), reverse=True):
+                        _bin = os.path.join(_root, _ver, "bin")
+                        if os.path.isdir(_bin):
+                            candidates.append(_bin)
+                            break
+            except OSError:
+                pass
+        for _d in candidates:
+            if os.path.isdir(_d):
+                try:
+                    os.add_dll_directory(_d)
+                except (OSError, AttributeError):
+                    pass
+
+    _add_rocm_dll_dirs()
+    del _add_rocm_dll_dirs, _ctypes
+
 # Ensure backend dir is on sys.path so _platform_compat is importable when
 # main.py is launched directly (e.g. `uvicorn main:app`).
 _backend_dir = str(_Path(__file__).parent)