fix windows hipblaslt check

2026-09-19 09:14:35 +02:00 · 2026-01-06 11:32:04 +09:00
parent 7f9cc66045
commit f4b592ac3a
1 changed files with 1 additions and 1 deletions
@@ -296,7 +296,7 @@ if sys.platform == "win32":

            agent = get_hip_agent()
            if torch.version.hip is not None and not agent.blaslt_supported:
-                torch.backends.cuda.preferred_blas_library('hipblas')
+                os.environ['DISABLE_ADDMM_CUDA_LT'] = '1'
                log.debug('ROCm: disabled hipBLASLt')
            if (agent.gfx_version & 0xFFF0) == 0x1200:
                # disable MIOpen for gfx120x