From 7f417dc598268c85ea60bc7383d8a27ddd224ff4 Mon Sep 17 00:00:00 2001 From: Chris Freehill Date: Tue, 3 Dec 2024 14:01:50 -0600 Subject: [PATCH] Support for generic ISAs in rocm_agent_enumerator Change-Id: Ie6cf2f1930e27afd6564250e108af1f10864946e --- rocm_agent_enumerator | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rocm_agent_enumerator b/rocm_agent_enumerator index 08a0596e99..2e77c2cd57 100755 --- a/rocm_agent_enumerator +++ b/rocm_agent_enumerator @@ -81,7 +81,7 @@ def staticVars(**kwargs): return func return deco -@staticVars(search_term=re.compile(r"gfx[0-9a-fA-F]+")) +@staticVars(search_term=re.compile(r"gfx[0-9a-fA-F]+(-[0-9a-fA-F]+)?(-generic)?")) def getGCNISA(line, match_from_beginning = False): if match_from_beginning is True: result = getGCNISA.search_term.match(line) @@ -92,7 +92,7 @@ def getGCNISA(line, match_from_beginning = False): return result.group(0) return None -@staticVars(search_name=re.compile(r"gfx[0-9a-fA-F]+(:[-+:\w]+)?")) +@staticVars(search_name=re.compile(r"gfx[0-9a-fA-F]+(-[0-9a-fA-F]+)?(-generic)?(:[-+:\w]+)?")) def getGCNArchName(line): result = getGCNArchName.search_name.search(line)