| @@ -108,7 +108,6 @@ class AicpuNodeTaskBase : public NodeTask { | |||||
| std::unique_ptr<TensorBuffer> copy_input_data_size_dev_; | std::unique_ptr<TensorBuffer> copy_input_data_size_dev_; | ||||
| std::unique_ptr<TensorBuffer> copy_input_src_dev_; | std::unique_ptr<TensorBuffer> copy_input_src_dev_; | ||||
| std::unique_ptr<TensorBuffer> copy_input_dst_dev_; | std::unique_ptr<TensorBuffer> copy_input_dst_dev_; | ||||
| bool need_sync_ = false; | |||||
| }; | }; | ||||
| class AicpuTfNodeTask : public AicpuNodeTaskBase { | class AicpuTfNodeTask : public AicpuNodeTaskBase { | ||||
| @@ -151,6 +150,8 @@ class AicpuTfNodeTask : public AicpuNodeTaskBase { | |||||
| std::unique_ptr<TensorBuffer> copy_ioaddr_dev_; | std::unique_ptr<TensorBuffer> copy_ioaddr_dev_; | ||||
| std::unique_ptr<TensorBuffer> copy_workspace_buf_; | std::unique_ptr<TensorBuffer> copy_workspace_buf_; | ||||
| bool need_sync_ = false; | |||||
| }; | }; | ||||
| class AicpuNodeTask : public AicpuNodeTaskBase { | class AicpuNodeTask : public AicpuNodeTaskBase { | ||||
| @@ -388,7 +388,7 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { | |||||
| domi::KernelDef *kernel_def = task_def->mutable_kernel(); | domi::KernelDef *kernel_def = task_def->mutable_kernel(); | ||||
| kernel_def->set_args(reinterpret_cast<const char *>(&args), args.head.length); | kernel_def->set_args(reinterpret_cast<const char *>(&args), args.head.length); | ||||
| kernel_def->set_args_size(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.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE; | ||||
| aicpu_ext_info.infoLen = sizeof(int32_t); | aicpu_ext_info.infoLen = sizeof(int32_t); | ||||
| memcpy_s(aicpu_ext_info.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); | memcpy_s(aicpu_ext_info.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); | ||||