fix offline debugger async dump segmentation fault issue

This commit is contained in:
Parastoo Ashtari 2021-09-29 12:29:31 -04:00
parent d94e2f7dcd
commit 6a568bfb5c
1 changed files with 4 additions and 2 deletions

View File

@ -1103,9 +1103,11 @@ std::vector<std::shared_ptr<TensorData>> DebugServices::ReadNeededDumpedTensors(
} else { } else {
dump_style_name += ".input"; dump_style_name += ".input";
} }
if (std::find(proto_to_dump.begin(), proto_to_dump.end(),
std::tuple<std::string, std::string>(orig_name, dump_style_name)) == proto_to_dump.end()) {
proto_to_dump.push_back(std::tuple<std::string, std::string>(orig_name, dump_style_name)); proto_to_dump.push_back(std::tuple<std::string, std::string>(orig_name, dump_style_name));
} }
}
if (!is_sync_mode_) { if (!is_sync_mode_) {
// convert all files in proto_to_dump to npy and add to pool of async file names // convert all files in proto_to_dump to npy and add to pool of async file names