From f8f987fed62b6958d8ba026fc84d7f6d6283df31 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Sun, 23 Feb 2025 01:19:40 +0900 Subject: [PATCH] zluda miopen temp --- modules/zluda_installer.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py index 47ee4a4ea..dc3d85e61 100644 --- a/modules/zluda_installer.py +++ b/modules/zluda_installer.py @@ -24,9 +24,11 @@ MIOpen_available = False path = os.path.abspath(os.environ.get('ZLUDA', '.zluda')) default_agent: Union[rocm.Agent, None] = None -nightly = os.environ.get("ZLUDA_NIGHTLY", "0") == "1" hipBLASLt_enabled = False +nightly = os.environ.get("ZLUDA_NIGHTLY", "0") == "1" +skip_arch_test = os.environ.get("ZLUDA_SKIP_ARCH_TEST", "0") == "1" + def set_default_agent(agent: rocm.Agent): global default_agent # pylint: disable=global-statement @@ -46,7 +48,7 @@ def set_default_agent(agent: rocm.Agent): hipBLASLt_enabled = hipBLASLt_available and ((not os.path.exists(path) and nightly) or os.path.exists(os.path.join(path, 'cublasLt.dll'))) global MIOpen_available # pylint: disable=global-statement - MIOpen_available = is_nightly and agent.gfx_version in (0x908, 0x90a, 0x940, 0x941, 0x942, 0x1030, 0x1100, 0x1101, 0x1102,) + MIOpen_available = is_nightly and (skip_arch_test or agent.gfx_version in (0x908, 0x90a, 0x940, 0x941, 0x942, 0x1030, 0x1100, 0x1101, 0x1102, 0x1150,)) def is_reinstall_needed() -> bool: # ZLUDA<3.8.7