From 8e6c104db91e4875db542725326e3dd09ed7d419 Mon Sep 17 00:00:00 2001 From: zhaozhixuan Date: Mon, 5 Jul 2021 21:53:27 +0800 Subject: [PATCH] Fix ut. --- ge/single_op/single_op_model.cc | 2 +- tests/ut/ge/single_op/single_op_model_unittest.cc | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/ge/single_op/single_op_model.cc b/ge/single_op/single_op_model.cc index f8831884..a5547b39 100755 --- a/ge/single_op/single_op_model.cc +++ b/ge/single_op/single_op_model.cc @@ -596,7 +596,7 @@ Status SingleOpModel::BuildTaskListForDynamicOp(StreamResource *stream_resource, const auto &atomic_task_def = task_defs.front(); AtomicOpTask *atomic_task = nullptr; GE_CHK_STATUS_RET_NOLOG(BuildAtomicTask(atomic_task_def, &atomic_task)); - atomic_task->InitAtomicAddrCleanIndices(); + GE_CHK_STATUS_RET_NOLOG(atomic_task->InitAtomicAddrCleanIndices()); tbe_task->SetAtomicTask(atomic_task); } single_op.op_task_.reset(tbe_task); diff --git a/tests/ut/ge/single_op/single_op_model_unittest.cc b/tests/ut/ge/single_op/single_op_model_unittest.cc index ded1b465..7b7a05d8 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -346,7 +346,6 @@ TEST_F(UtestSingleOpModel, build_dynamic_task) { model.model_helper_.model_ = ge_model; auto op_desc = std::make_shared("add", "Add"); AttrUtils::SetStr(op_desc, TVM_ATTR_NAME_MAGIC, "RT_DEV_BINARY_MAGIC_ELF"); - AttrUtils::SetBool(op_desc, kAttrSupportDynamicShape, true); std::vector kernelBin; TBEKernelPtr tbe_kernel = std::make_shared("name/Add", std::move(kernelBin)); op_desc->SetExtAttr(ge::OP_EXTATTR_NAME_TBE_KERNEL, tbe_kernel);