|
|
|
@ -137,18 +137,20 @@ bool DataDumpParser::NeedDump(const std::string &op_full_name) const {
|
|
|
|
|
return iter != kernel_map_.end();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool DataDumpParser::IsConfigExist(const nlohmann::json &dump_settings) const {
|
|
|
|
|
if (dump_settings.find(kConfigDumpMode) == dump_settings.end() ||
|
|
|
|
|
dump_settings.find(kConfigNetName) == dump_settings.end() ||
|
|
|
|
|
dump_settings.find(kConfigOpDebugMode) == dump_settings.end() ||
|
|
|
|
|
dump_settings.find(kConfigIteration) == dump_settings.end() ||
|
|
|
|
|
dump_settings.find(kConfigKernels) == dump_settings.end()) {
|
|
|
|
|
MS_LOG(ERROR) << "[DataDump] DumpSettings keys are not exist.";
|
|
|
|
|
bool CheckConfigKey(const nlohmann::json &dump_settings, const std::string &key) {
|
|
|
|
|
if (dump_settings.find(key) == dump_settings.end()) {
|
|
|
|
|
MS_LOG(ERROR) << "[DataDump] DumpSettings key:" << key << " is not exist.";
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool DataDumpParser::IsConfigExist(const nlohmann::json &dump_settings) const {
|
|
|
|
|
return CheckConfigKey(dump_settings, kConfigDumpMode) && CheckConfigKey(dump_settings, kConfigNetName) &&
|
|
|
|
|
CheckConfigKey(dump_settings, kConfigOpDebugMode) && CheckConfigKey(dump_settings, kConfigIteration) &&
|
|
|
|
|
CheckConfigKey(dump_settings, kConfigKernels);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool DataDumpParser::ParseDumpSetting(const nlohmann::json &dump_settings) {
|
|
|
|
|
auto mode = dump_settings.at(kConfigDumpMode);
|
|
|
|
|
auto op_debug_mode = dump_settings.at(kConfigOpDebugMode);
|
|
|
|
|