Browse Source

fix ut

pull/2039/head
guopeian 3 years ago
parent
commit
a5210e2701
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
  2. +1
    -1
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 2
- 1
ge/hybrid/node_executor/aicpu/aicpu_node_executor.h View File

@@ -108,7 +108,6 @@ class AicpuNodeTaskBase : public NodeTask {
std::unique_ptr<TensorBuffer> copy_input_data_size_dev_;
std::unique_ptr<TensorBuffer> copy_input_src_dev_;
std::unique_ptr<TensorBuffer> copy_input_dst_dev_;
bool need_sync_ = false;
};

class AicpuTfNodeTask : public AicpuNodeTaskBase {
@@ -151,6 +150,8 @@ class AicpuTfNodeTask : public AicpuNodeTaskBase {
std::unique_ptr<TensorBuffer> copy_ioaddr_dev_;

std::unique_ptr<TensorBuffer> copy_workspace_buf_;

bool need_sync_ = false;
};

class AicpuNodeTask : public AicpuNodeTaskBase {


+ 1
- 1
tests/ut/ge/single_op/single_op_model_unittest.cc View File

@@ -388,7 +388,7 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) {
domi::KernelDef *kernel_def = task_def->mutable_kernel();
kernel_def->set_args(reinterpret_cast<const char *>(&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));


Loading…
Cancel
Save