| @@ -1112,7 +1112,15 @@ Status AiCpuCCTask::LaunchKernel(const std::vector<GeTensorDesc> &input_desc, | |||||
| std::vector<DataBuffer> &output_buffers, | std::vector<DataBuffer> &output_buffers, | ||||
| rtStream_t stream) { | rtStream_t stream) { | ||||
| GE_CHK_STATUS_RET_NOLOG(UpdateExtInfo(input_desc, output_desc, stream)); | GE_CHK_STATUS_RET_NOLOG(UpdateExtInfo(input_desc, output_desc, stream)); | ||||
| GE_CHK_STATUS_RET_NOLOG(UpdateIoAddr(input_buffers, output_buffers)); | |||||
| if (unknown_type_ == DEPEND_COMPUTE) { | |||||
| std::vector<DataBuffer> summary_buffers; | |||||
| for (size_t i = 0; i < num_outputs_; ++i) { | |||||
| summary_buffers.emplace_back(output_summary_[i], sizeof(aicpu::FWKAdapter::ResultSummary), false); | |||||
| } | |||||
| GE_CHK_STATUS_RET_NOLOG(UpdateIoAddr(input_buffers, summary_buffers)); | |||||
| } else { | |||||
| GE_CHK_STATUS_RET_NOLOG(UpdateIoAddr(input_buffers, output_buffers)); | |||||
| } | |||||
| GE_CHK_STATUS_RET_NOLOG(LaunchKernel(stream)); | GE_CHK_STATUS_RET_NOLOG(LaunchKernel(stream)); | ||||
| if (unknown_type_ == DEPEND_SHAPE_RANGE) { | if (unknown_type_ == DEPEND_SHAPE_RANGE) { | ||||
| GE_CHK_RT_RET(rtStreamSynchronize(stream)); | GE_CHK_RT_RET(rtStreamSynchronize(stream)); | ||||