Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

rocm get version exception handler

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-02-21 14:57:12 +01:00
parent d79d4edddb
commit a96b5234d7
1 changed files with 23 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/rocm.py
+23 -19
View File
@@ -209,27 +209,30 @@ def find() -> ROCmEnvironment | None:
def get_version() -> str:
if isinstance(environment, ROCmEnvironment):
# We don't load the hip library that will not be used by PyTorch.
if sys.platform == "win32":
# ROCm is system-wide installed. Assume the version is the folder name. (e.g. C:\Program Files\AMD\ROCm\6.4)
# hipconfig requires Perl
return os.path.basename(environment.path) or os.path.basename(os.path.dirname(environment.path))
try:
if isinstance(environment, ROCmEnvironment):
# We don't load the hip library that will not be used by PyTorch.
if sys.platform == "win32":
# ROCm is system-wide installed. Assume the version is the folder name. (e.g. C:\Program Files\AMD\ROCm\6.4)
# hipconfig requires Perl
return os.path.basename(environment.path) or os.path.basename(os.path.dirname(environment.path))
else:
arr = spawn("hipconfig --version", cwd=os.path.join(environment.path, 'bin')).split(".")
return f'{arr[0]}.{arr[1]}' if len(arr) >= 2 else None
elif isinstance(environment, PythonPackageEnvironment):
# If rocm-sdk package is installed, the hip library may be used by PyTorch.
ver = ctypes.c_int()
environment.hip.hipRuntimeGetVersion(ctypes.byref(ver))
major = ver.value // 10000000
minor = (ver.value // 100000) % 100
#patch = version.value % 100000
return f"{major}.{minor}"
else:
arr = spawn("hipconfig --version", cwd=os.path.join(environment.path, 'bin')).split(".")
return f'{arr[0]}.{arr[1]}' if len(arr) >= 2 else None
elif isinstance(environment, PythonPackageEnvironment):
# If rocm-sdk package is installed, the hip library may be used by PyTorch.
ver = ctypes.c_int()
environment.hip.hipRuntimeGetVersion(ctypes.byref(ver))
major = ver.value // 10000000
minor = (ver.value // 100000) % 100
#patch = version.value % 100000
return f"{major}.{minor}"
else:
return None
except Exception as e:
log.error(f'ROCm: failed to get version: {e}')
return None
def get_flash_attention_command(agent: Agent) -> str:
default = "git+https://github.com/ROCm/flash-attention"
if agent.gfx_version >= 0x1100 and agent.gfx_version < 0x1200 and os.environ.get("FLASH_ATTENTION_USE_TRITON_ROCM", "false").lower() != "true":
@@ -242,7 +245,8 @@ def get_flash_attention_command(agent: Agent) -> str:
def refresh():
global rocm_sdk, environment, blaslt_tensile_libpath, is_installed, version # pylint: disable=global-statement
try:
import rocm_sdk
import rocm_sdk as rocm_sdk_module
rocm_sdk = rocm_sdk_module
environment = PythonPackageEnvironment(rocm_sdk)
try:
target_family = rocm_sdk._dist_info.determine_target_family() # pylint: disable=protected-access