skip rocm.conceal if rocm torch is installed

2026-09-19 17:24:32 +02:00 · 2025-09-28 17:34:47 +09:00
parent 95ee0ed29c
commit e8a507fe2a
1 changed files with 4 additions and 2 deletions
@@ -977,8 +977,10 @@ def check_torch():
            if not args.ignore:
                sys.exit(1)
    if rocm.is_installed:
-        if sys.platform == "win32": # CPU, DirectML, ZLUDA
-            rocm.conceal()
+        if sys.platform == "win32":
+            import torch
+            if torch.version.hip is None: # CPU, DirectML, ZLUDA
+                rocm.conceal()
        elif rocm.is_wsl: # ROCm WSL
            rocm.preload_hsa_runtime()
    if args.version: