diff options
Diffstat (limited to 'ethosu')
-rw-r--r-- | ethosu/vela/architecture_features.py | 28 |
1 files changed, 17 insertions, 11 deletions
diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py index 64005bf5..f7dcc8ce 100644 --- a/ethosu/vela/architecture_features.py +++ b/ethosu/vela/architecture_features.py @@ -802,16 +802,22 @@ class ArchitectureFeatures: return result +# Cache for default arch instances, as these are expensive to create +default_arch_cache = dict() + + def create_default_arch(accelerator: Accelerator) -> ArchitectureFeatures: """Creates architecture features object using default settings""" - return ArchitectureFeatures( - vela_config_files=None, - accelerator_config=accelerator.value, - system_config=ArchitectureFeatures.DEFAULT_CONFIG, - memory_mode=ArchitectureFeatures.DEFAULT_CONFIG, - override_block_config=None, - block_config_limit=None, - max_blockdep=ArchitectureFeatures.MAX_BLOCKDEP, - weight_estimation_scaling=1.0, - verbose_config=False, - ) + if accelerator not in default_arch_cache: + default_arch_cache[accelerator] = ArchitectureFeatures( + vela_config_files=None, + accelerator_config=accelerator.value, + system_config=ArchitectureFeatures.DEFAULT_CONFIG, + memory_mode=ArchitectureFeatures.DEFAULT_CONFIG, + override_block_config=None, + block_config_limit=None, + max_blockdep=ArchitectureFeatures.MAX_BLOCKDEP, + weight_estimation_scaling=1.0, + verbose_config=False, + ) + return default_arch_cache[accelerator] |