forked from huawei/mindspore2022
138 lines
3.9 KiB
C++
138 lines
3.9 KiB
C++
/**
|
|
* Copyright 2020 Huawei Technologies Co., Ltd
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef MINDSPORE_LITE_TOOLS_CONVERTER_CONVERTER_FLAGS_H
|
|
#define MINDSPORE_LITE_TOOLS_CONVERTER_CONVERTER_FLAGS_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include "include/api/format.h"
|
|
#include "include/registry/converter_context.h"
|
|
#include "tools/common/flag_parser.h"
|
|
#include "ir/dtype/type_id.h"
|
|
#include "schema/inner/model_generated.h"
|
|
#include "tools/converter/preprocess/preprocess_param.h"
|
|
#include "tools/converter/quantizer/quant_params.h"
|
|
#include "tools/converter/adapter/acl/common/acl_types.h"
|
|
#include "micro/coder/config.h"
|
|
|
|
namespace mindspore {
|
|
namespace lite {
|
|
class ConfigFileParser;
|
|
} // namespace lite
|
|
namespace converter {
|
|
using mindspore::schema::QuantType;
|
|
enum ParallelSplitType { SplitNo = 0, SplitByUserRatio = 1, SplitByUserAttr = 2 };
|
|
constexpr auto kMaxSplitRatio = 10;
|
|
constexpr auto kComputeRate = "computeRate";
|
|
constexpr auto kSplitDevice0 = "device0";
|
|
constexpr auto kSplitDevice1 = "device1";
|
|
constexpr size_t kEncMaxLen = 16;
|
|
struct ParallelSplitConfig {
|
|
ParallelSplitType parallel_split_type_ = SplitNo;
|
|
std::vector<int64_t> parallel_compute_rates_;
|
|
std::vector<std::string> parallel_devices_;
|
|
};
|
|
|
|
class Flags : public virtual mindspore::lite::FlagParser {
|
|
public:
|
|
Flags();
|
|
|
|
~Flags() override;
|
|
|
|
int InitInputOutputDataType();
|
|
|
|
int InitFmk();
|
|
|
|
int InitTrainModel();
|
|
|
|
int InitConfigFile();
|
|
|
|
int InitInTensorShape() const;
|
|
|
|
int InitGraphInputFormat();
|
|
|
|
int InitExtendedIntegrationInfo(const lite::ConfigFileParser &config_file_parser);
|
|
|
|
int InitEncrypt();
|
|
|
|
int InitPreInference();
|
|
|
|
int InitSaveFP16();
|
|
|
|
void InitAclDefaultOption();
|
|
|
|
int InitExportMindIR();
|
|
|
|
int Init(int argc, const char **argv);
|
|
|
|
int PreInit(int argc, const char **argv);
|
|
|
|
std::string modelFile;
|
|
std::string outputFile;
|
|
std::string fmkIn;
|
|
FmkType fmk;
|
|
std::string weightFile;
|
|
TypeId inputDataType;
|
|
TypeId outputDataType;
|
|
std::string saveFP16Str = "off";
|
|
bool saveFP16 = false;
|
|
std::string inputDataTypeStr;
|
|
std::string outputDataTypeStr;
|
|
ParallelSplitConfig parallel_split_config_{};
|
|
std::string configFile;
|
|
std::string trainModelIn;
|
|
bool trainModel = false;
|
|
std::vector<std::string> pluginsPath;
|
|
bool disableFusion = false;
|
|
std::string inTensorShape;
|
|
std::string dec_key = "";
|
|
std::string dec_mode = "AES-GCM";
|
|
std::string graphInputFormatStr;
|
|
std::string device;
|
|
mindspore::Format graphInputFormat = mindspore::NHWC;
|
|
std::string encKeyStr;
|
|
std::string encMode = "AES-GCM";
|
|
std::string inferStr;
|
|
std::string exportMindIR;
|
|
#ifdef ENABLE_OPENSSL
|
|
std::string encryptionStr = "true";
|
|
bool encryption = true;
|
|
#else
|
|
std::string encryptionStr = "false";
|
|
bool encryption = false;
|
|
#endif
|
|
bool infer = false;
|
|
unsigned char encKey[kEncMaxLen];
|
|
size_t keyLen = 0;
|
|
bool export_mindir = false;
|
|
|
|
lite::quant::CommonQuantParam commonQuantParam;
|
|
lite::quant::MixedBitWeightQuantParam mixedBitWeightQuantParam;
|
|
lite::quant::FullQuantParam fullQuantParam;
|
|
lite::preprocess::DataPreProcessParam dataPreProcessParam;
|
|
lite::acl::AclModelOptionCfg aclModelOptionCfgParam;
|
|
lite::micro::MicroParam microParam;
|
|
};
|
|
|
|
bool CheckOfflineParallelConfig(const std::string &file, ParallelSplitConfig *parallel_split_config);
|
|
|
|
std::string GetStrFromConfigFile(const std::string &file, const std::string &target_key);
|
|
} // namespace converter
|
|
} // namespace mindspore
|
|
|
|
#endif
|