From a5210e2701579e00ce52e2dd1ab2fdf3c3b7d2ee Mon Sep 17 00:00:00 2001 From: guopeian Date: Mon, 19 Jul 2021 15:25:13 +0800 Subject: [PATCH] fix ut --- ge/hybrid/node_executor/aicpu/aicpu_node_executor.h | 3 ++- tests/ut/ge/single_op/single_op_model_unittest.cc | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h index f9001f83..37dd4d0b 100644 --- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h +++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h @@ -108,7 +108,6 @@ class AicpuNodeTaskBase : public NodeTask { std::unique_ptr copy_input_data_size_dev_; std::unique_ptr copy_input_src_dev_; std::unique_ptr copy_input_dst_dev_; - bool need_sync_ = false; }; class AicpuTfNodeTask : public AicpuNodeTaskBase { @@ -151,6 +150,8 @@ class AicpuTfNodeTask : public AicpuNodeTaskBase { std::unique_ptr copy_ioaddr_dev_; std::unique_ptr copy_workspace_buf_; + + bool need_sync_ = false; }; class AicpuNodeTask : public AicpuNodeTaskBase { 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 28a8418c..04d4317b 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -388,7 +388,7 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { domi::KernelDef *kernel_def = task_def->mutable_kernel(); kernel_def->set_args(reinterpret_cast(&args), args.head.length); kernel_def->set_args_size(args.head.length); - AicpuExtInfo aicpu_ext_info; + ge::hybrid::AicpuExtInfo aicpu_ext_info; aicpu_ext_info.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE; aicpu_ext_info.infoLen = sizeof(int32_t); memcpy_s(aicpu_ext_info.infoMsg, sizeof(int32_t), &type, sizeof(int32_t));