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__.py30
1 files changed, 13 insertions, 17 deletions
diff --git a/src/mlia/backend/corstone/__init__.py b/src/mlia/backend/corstone/__init__.py
index 36f74ee..b59ab65 100644
--- a/src/mlia/backend/corstone/__init__.py
+++ b/src/mlia/backend/corstone/__init__.py
@@ -7,24 +7,20 @@ from mlia.backend.config import System
from mlia.backend.registry import registry
from mlia.core.common import AdviceCategory
-registry.register(
- "Corstone-300",
- BackendConfiguration(
- supported_advice=[AdviceCategory.PERFORMANCE, AdviceCategory.OPTIMIZATION],
- supported_systems=[System.LINUX_AMD64],
- backend_type=BackendType.CUSTOM,
- ),
-)
-registry.register(
- "Corstone-310",
- BackendConfiguration(
- supported_advice=[AdviceCategory.PERFORMANCE, AdviceCategory.OPTIMIZATION],
- supported_systems=[System.LINUX_AMD64],
- backend_type=BackendType.CUSTOM,
- ),
-)
+# List of mutually exclusive Corstone backends ordered by priority
+CORSTONE_PRIORITY = ("Corstone-310", "Corstone-300")
+
+for corstone_name in CORSTONE_PRIORITY:
+ registry.register(
+ corstone_name,
+ BackendConfiguration(
+ supported_advice=[AdviceCategory.PERFORMANCE, AdviceCategory.OPTIMIZATION],
+ supported_systems=[System.LINUX_AMD64],
+ backend_type=BackendType.CUSTOM,
+ ),
+ )
def is_corstone_backend(backend_name: str) -> bool:
"""Check if backend belongs to Corstone."""
- return backend_name in ["Corstone-300", "Corstone-310"]
+ return backend_name in CORSTONE_PRIORITY