diff options
| author | Nathan Bailey <nathan.bailey@arm.com> | 2024-01-16 16:39:06 +0000 |
|---|---|---|
| committer | Nathan Bailey <nathan.bailey@arm.com> | 2024-02-09 14:31:46 +0000 |
| commit | d08513a72e7fbf0626c3d69b9c4cc7056b3da4ae (patch) | |
| tree | 3d332d70dbc57b6b7255bb4b429b322ad90bb2c1 /src/mlia/backend/vela/compiler.py | |
| parent | be7bab89eb8ace6ab6a83687354beab156afb716 (diff) | |
| download | mlia-d08513a72e7fbf0626c3d69b9c4cc7056b3da4ae.tar.gz | |
feat: Integrate Vela's per-layer performance estimates
Resolves: MLIA-1055, MLIA-1056, MLIA-1057
Signed-off-by: Nathan Bailey <nathan.bailey@arm.com>
Change-Id: Id573cec94e4a69117051dcd5175f383c0955d890
Diffstat (limited to 'src/mlia/backend/vela/compiler.py')
| -rw-r--r-- | src/mlia/backend/vela/compiler.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/mlia/backend/vela/compiler.py b/src/mlia/backend/vela/compiler.py index b591056..fe9e365 100644 --- a/src/mlia/backend/vela/compiler.py +++ b/src/mlia/backend/vela/compiler.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2024, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Vela compiler wrapper module.""" from __future__ import annotations @@ -90,7 +90,7 @@ class VelaCompilerOptions: # pylint: disable=too-many-instance-attributes tensor_allocator: TensorAllocatorType = "HillClimb" cpu_tensor_alignment: int = Tensor.AllocationQuantum optimization_strategy: OptimizationStrategyType = "Performance" - output_dir: str = "output" + output_dir: Path = Path("output") recursion_limit: int = 1000 @@ -251,6 +251,7 @@ class VelaCompiler: # pylint: disable=too-many-instance-attributes verbose_register_command_stream=False, verbose_operators=False, verbose_weights=False, + verbose_performance=True, show_cpu_operations=False, tensor_allocator=self.tensor_allocator, timing=False, @@ -258,6 +259,10 @@ class VelaCompiler: # pylint: disable=too-many-instance-attributes cpu_tensor_alignment=self.cpu_tensor_alignment, ) + def return_compiler_options(self) -> CompilerOptions: + """Return CompilerOptions instance for test purposes.""" + return self._compiler_options() + def resolve_compiler_config( vela_compiler_options: VelaCompilerOptions, |