Support for generic ISAs in rocm_agent_enumerator

Change-Id: Ie6cf2f1930e27afd6564250e108af1f10864946e
Этот коммит содержится в:
Chris Freehill
2024-12-03 14:01:50 -06:00
родитель 22ed708c39
Коммит 7f417dc598
+2 -2
Просмотреть файл
@@ -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)