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