diff --git a/rocm_agent_enumerator b/rocm_agent_enumerator index 6264a5f..b357fbc 100755 --- a/rocm_agent_enumerator +++ b/rocm_agent_enumerator @@ -92,7 +92,7 @@ def getGCNISA(line, match_from_beginning = False): return result.group(0) return None -@staticVars(search_name=re.compile("gfx[0-9a-fA-F]+:[-+:\w]+")) +@staticVars(search_name=re.compile("gfx[0-9a-fA-F]+(:[-+:\w]+)?")) def getGCNArchName(line): result = getGCNArchName.search_name.search(line)