aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/backend/corstone/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/backend/corstone/__init__.py')
-rw-r--r--src/mlia/backend/corstone/__init__.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/src/mlia/backend/corstone/__init__.py b/src/mlia/backend/corstone/__init__.py
index b59ab65..5aa688b 100644
--- a/src/mlia/backend/corstone/__init__.py
+++ b/src/mlia/backend/corstone/__init__.py
@@ -12,15 +12,19 @@ CORSTONE_PRIORITY = ("Corstone-310", "Corstone-300")
for corstone_name in CORSTONE_PRIORITY:
registry.register(
- corstone_name,
+ corstone_name.lower(),
BackendConfiguration(
supported_advice=[AdviceCategory.PERFORMANCE, AdviceCategory.OPTIMIZATION],
supported_systems=[System.LINUX_AMD64],
backend_type=BackendType.CUSTOM,
),
+ pretty_name=corstone_name,
)
def is_corstone_backend(backend_name: str) -> bool:
"""Check if backend belongs to Corstone."""
- return backend_name in CORSTONE_PRIORITY
+ return any(
+ name in CORSTONE_PRIORITY
+ for name in (backend_name, registry.pretty_name(backend_name))
+ )