• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /**
2  * Copyright 2023 Huawei Technologies Co., Ltd
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "tools/converter/config_parser/cpu_option_param_parser.h"
18 
19 namespace mindspore {
20 namespace lite {
ParseCpuOptionCfg(const CpuOptionCfgString & cpu_option_string,CpuOptionCfg * cpu_option_cfg)21 STATUS CpuOptionParamParser::ParseCpuOptionCfg(const CpuOptionCfgString &cpu_option_string,
22                                                CpuOptionCfg *cpu_option_cfg) {
23   if (cpu_option_string.architecture.empty() || cpu_option_string.instruction.empty()) {
24     return RET_OK;
25   }
26 
27   if (cpu_option_string.architecture != "ARM64") {
28     MS_LOG(ERROR) << "cpu instruction only supported ARM64. But get " << cpu_option_string.architecture;
29     return RET_INPUT_PARAM_INVALID;
30   }
31 
32   if (cpu_option_string.instruction != "SIMD_DOT") {
33     MS_LOG(ERROR) << "cpu instruction only supported SIMD_DOT. But get " << cpu_option_string.instruction;
34     return RET_INPUT_PARAM_INVALID;
35   }
36   cpu_option_cfg->instruction = cpu_option_string.instruction;
37   cpu_option_cfg->architecture = cpu_option_string.architecture;
38   return RET_OK;
39 }
40 }  // namespace lite
41 }  // namespace mindspore
42