diff options
Diffstat (limited to 'src/mlia/backend/corstone/__init__.py')
-rw-r--r-- | src/mlia/backend/corstone/__init__.py | 8 |
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)) + ) |