Fix bug.

vladmandic · Jan 11, 2025 · 389bb78 · 389bb78
1 parent e43d8e9
commit 389bb78
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/rocm.py b/modules/rocm.py
@@ -212,9 +212,10 @@ def get_flash_attention_command(agent: Agent):
     is_wsl: bool = os.environ.get('WSL_DISTRO_NAME', 'unknown' if spawn('wslpath -w /') else None) is not None
     version_torch = get_version_torch()
 path = find()
-blaslt_tensile_libpath = os.environ.get("HIPBLASLT_TENSILE_LIBPATH", os.path.join(path, "bin" if sys.platform == "win32" else "lib", "hipblaslt", "library"))
+blaslt_tensile_libpath = None
 is_installed = False
 version = None
 if path is not None:
+    blaslt_tensile_libpath = os.environ.get("HIPBLASLT_TENSILE_LIBPATH", os.path.join(path, "bin" if sys.platform == "win32" else "lib", "hipblaslt", "library"))
     is_installed = True
     version = get_version()