From f51a80f3abc508df08cde3e08c60a7510a6f5282 Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Sat, 9 Jan 2021 19:32:36 +0800 Subject: [PATCH] bugfix for l1 data dump --- ge/graph/load/new_model_manager/davinci_model.cc | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ge/graph/load/new_model_manager/davinci_model.cc b/ge/graph/load/new_model_manager/davinci_model.cc index 44f46785..c531fe13 100755 --- a/ge/graph/load/new_model_manager/davinci_model.cc +++ b/ge/graph/load/new_model_manager/davinci_model.cc @@ -4160,7 +4160,12 @@ Status DavinciModel::InitL1DataDumperArgs() { GE_CHK_RT_RET(rtMalloc(&l1_fusion_addr_, kDumpL1FusionOpMByteSize, RT_MEMORY_DDR)); // send l1fusion dump addr to rts - GE_CHK_RT_RET(rtDumpAddrSet(rt_model_handle_, l1_fusion_addr_, kDumpL1FusionOpMByteSize, kDumpFlagOfL1Fusion)); + if (rtDumpAddrSet(rt_model_handle_, l1_fusion_addr_, kDumpL1FusionOpMByteSize, kDumpFlagOfL1Fusion) != + RT_ERROR_NONE) { + GELOGE(FAILED, "Call rtDumpAddrSet failed"); + GE_CHK_RT(rtFree(l1_fusion_addr_)); + return FAILED; + } // set addr for l1 data dump data_dumper_.SetL1FusionAddr(l1_fusion_addr_);