diff options
Diffstat (limited to 'src/mlia/backend')
| -rw-r--r-- | src/mlia/backend/vela/compiler.py | 4 | ||||
| -rw-r--r-- | src/mlia/backend/vela/performance.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/src/mlia/backend/vela/compiler.py b/src/mlia/backend/vela/compiler.py index b62df24..afad05b 100644 --- a/src/mlia/backend/vela/compiler.py +++ b/src/mlia/backend/vela/compiler.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Vela compiler wrapper module.""" from __future__ import annotations @@ -268,7 +268,7 @@ def optimize_model( ) -> None: """Optimize model and return it's path after optimization.""" logger.debug( - "Optimize model %s for device %s", + "Optimize model %s for target %s", model_path, compiler_options.accelerator_config, ) diff --git a/src/mlia/backend/vela/performance.py b/src/mlia/backend/vela/performance.py index ccd2f6f..e545b85 100644 --- a/src/mlia/backend/vela/performance.py +++ b/src/mlia/backend/vela/performance.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Vela performance module.""" from __future__ import annotations @@ -42,7 +42,7 @@ class PerformanceMetrics: # pylint: disable=too-many-instance-attributes def estimate_performance( model_path: Path, compiler_options: VelaCompilerOptions ) -> PerformanceMetrics: - """Return performance estimations for the model/device. + """Return performance estimations for the model/target. Logic for this function comes from Vela module stats_writer.py """ |