From 283fc37232d11e7f1c233e16a82b67c3ebbfaeaa Mon Sep 17 00:00:00 2001 From: wqtshg Date: Tue, 13 Apr 2021 22:02:47 +0800 Subject: [PATCH] modify dump bug --- .../ut/ge/graph/load/data_dumper_unittest.cc | 33 +++++++++++++++++++ 1 file changed, 33 insertions(+) diff --git a/tests/ut/ge/graph/load/data_dumper_unittest.cc b/tests/ut/ge/graph/load/data_dumper_unittest.cc index 68040bf1..8f46e989 100644 --- a/tests/ut/ge/graph/load/data_dumper_unittest.cc +++ b/tests/ut/ge/graph/load/data_dumper_unittest.cc @@ -38,6 +38,19 @@ std::vector stub_get_output_addrs(const RuntimeParam &model_param, Const res.emplace_back(reinterpret_cast(23333)); return res; } + +static ge::OpDescPtr CreateOpDesc(string name = "", string type = "") { + auto op_desc = std::make_shared(name, type); + op_desc->SetStreamId(0); + op_desc->SetId(0); + + op_desc->SetWorkspace({}); + op_desc->SetWorkspaceBytes({}); + op_desc->SetInputOffset({}); + op_desc->SetOutputOffset({100, 200}); + return op_desc; +} + /* TEST_F(UtestDataDumper, LoadDumpInfo_no_output_addrs_fail) { RuntimeParam rts_param; @@ -63,4 +76,24 @@ TEST_F(UtestDataDumper, UnloadDumpInfo_success) { Status ret = data_dumper.UnloadDumpInfo(); EXPECT_EQ(ret, SUCCESS); } + +TEST_F(UtestDataDumper, DumpOutputWithTask_success) { + RuntimeParam rts_param; + DataDumper data_dumper(&rts_param); + data_dumper.SetModelName("test"); + data_dumper.SetModelId(2333); + + aicpu::dump::Task task; + OpDescPtr op_desc = CreateOpDesc("conv", CONVOLUTION); + GeTensorDesc tensor_0(GeShape(), FORMAT_NCHW, DT_FLOAT); + GeTensorDesc tensor_1(GeShape(), FORMAT_NCHW, DT_FLOAT); + int32_t calc_type = 1; + ge::AttrUtils::SetInt(tensor_1, ATTR_NAME_MEMORY_SIZE_CALC_TYPE, calc_type); + op_desc->AddOutputDesc(tensor_0); + op_desc->AddOutputDesc(tensor_1); + DataDumper::InnerDumpInfo inner_dump_info; + inner_dump_info.op = op_desc; + Status ret = data_dumper.DumpOutputWithTask(inner_dump_info, task); + EXPECT_EQ(ret, SUCCESS); +} } // namespace ge