windows gfx120x disable MIOpen

2026-09-20 01:31:13 +02:00 · 2025-11-03 00:36:54 +09:00
parent 8535efc14a
commit 57e8da7a36
1 changed files with 6 additions and 0 deletions
@@ -276,6 +276,12 @@ if sys.platform == "win32":
        try:
            import torch
            import numpy as np
+            from modules.devices import get_optimal_device
+
+            gfx_version = Agent.parse_gfx_version(getattr(torch.cuda.get_device_properties(get_optimal_device()), "gcnArchName", "gfx0000"))
+            if (gfx_version & 0xFFF0) == 0x1200:
+                # disable MIOpen for gfx120x
+                torch.backends.cudnn.enabled = False

            original_cholesky_ex = torch.linalg.cholesky_ex
            @wraps(original_cholesky_ex)