Skip to content

Commit 0f81550

Browse files
authored
[Debug] Record debug info for exporting op (#1568)
1 parent b499045 commit 0f81550

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddle2onnx/mapper/exporter.cc

+4
Original file line numberDiff line numberDiff line change
@@ -749,6 +749,8 @@ void ModelExporter::ExportOp(const PaddlePirParser& pir_parser,
749749
int64_t op_id,
750750
bool if_in_subblock,
751751
bool verbose) {
752+
P2OLogger(verbose_) << "Start export " << op->name()
753+
<< ", op_id: " << op->id() << std::endl;
752754
auto mapper =
753755
MapperHelper::Get()->CreateMapper(convert_pir_op_name(op->name()),
754756
pir_parser,
@@ -758,6 +760,8 @@ void ModelExporter::ExportOp(const PaddlePirParser& pir_parser,
758760
mapper->deploy_backend = deploy_backend_;
759761
mapper->Run();
760762
delete mapper;
763+
P2OLogger(verbose_) << "Finish export " << op->name()
764+
<< ", op_id: " << op->id() << std::endl;
761765
}
762766

763767
void ModelExporter::CovertCustomOps(const PaddleParser& parser,

0 commit comments

Comments
 (0)