From c13b8d2712c1ea6e24edfe183d417cfbfb40f266 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 6 Oct 2024 09:11:26 -0400 Subject: [PATCH] installer fixes Signed-off-by: Vladimir Mandic --- installer.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/installer.py b/installer.py index f13306071..f77398c69 100644 --- a/installer.py +++ b/installer.py @@ -595,7 +595,11 @@ def install_rocm_zluda(): if device is None: log.debug('ROCm: HSA_OVERRIDE_GFX_VERSION auto config skipped') else: - os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', device.get_gfx_version()) + gfx_ver = device.get_gfx_version() + if gfx_ver is not None: + os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', gfx_ver) + else: + log.warning('ROCm: hsa version detect failed') return torch_command @@ -902,7 +906,7 @@ def install_extensions(force=False): commit = extensions_commit.get(os.path.basename(ext), None) if commit is not None: log.debug(f'Extension force: name={ext} commit={commit}') - res = git(f'checkout {commit}', os.path.join(folder, ext)) + res.append(git(f'checkout {commit}', os.path.join(folder, ext))) run_extension_installer(os.path.join(folder, ext)) pkg_resources._initialize_master_working_set() # pylint: disable=protected-access try: