Skip to content

Commit 336d526

Browse files
jatinwadhwa921ankitm3k
authored andcommitted
[OVEP] Fix for dumping the model in correct format (#616)
1 parent 1108efa commit 336d526

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

onnxruntime/core/providers/openvino/backend_manager.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -331,6 +331,7 @@ static void DumpOpenVINOEPModel(const std::filesystem::path& onnx_model_path_nam
331331
if (dash != std::string::npos) {
332332
auto new_name = model_name.stem().string() + subgraph_name.substr(dash, std::string::npos);
333333
model_name.replace_filename(new_name);
334+
model_name.replace_extension(".onnx");
334335
}
335336

336337
std::fstream dump(model_name, std::ios::out | std::ios::trunc | std::ios::binary);

0 commit comments

Comments
 (0)