1#!/usr/bin/env python3 2# -*- coding: utf-8 -*- 3 4# Copyright (c) 2021-2024 Huawei Device Co., Ltd. 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16# 17 18from functools import cached_property 19from typing import Dict, List 20 21from runner.options.decorator_value import value, _to_bool, _to_int, _to_str 22 23 24class ArkAotOptions: 25 __DEFAULT_TIMEOUT = 600 26 27 def __str__(self) -> str: 28 return _to_str(self, 1) 29 30 def to_dict(self) -> Dict[str, object]: 31 return { 32 "enable": self.enable, 33 "timeout": self.timeout, 34 "aot-args": self.aot_args, 35 } 36 37 @cached_property 38 @value(yaml_path="ark_aot.enable", cli_name="aot", cast_to_type=_to_bool) 39 def enable(self) -> bool: 40 return False 41 42 @cached_property 43 @value(yaml_path="ark_aot.timeout", cli_name="paoc_timeout", cast_to_type=_to_int) 44 def timeout(self) -> int: 45 return ArkAotOptions.__DEFAULT_TIMEOUT 46 47 @cached_property 48 @value(yaml_path="ark_aot.aot-args", cli_name="aot_args") 49 def aot_args(self) -> List[str]: 50 return [] 51 52 def get_command_line(self) -> str: 53 _aot_args = [f'--aot-args="{arg}"' for arg in self.aot_args] 54 options = [ 55 '--aot' if self.enable else '', 56 f'--paoc-timeout={self.timeout}' if self.timeout != ArkAotOptions.__DEFAULT_TIMEOUT else '', 57 ' '.join(_aot_args) 58 ] 59 return ' '.join(options) 60