|
|
@@ -1112,7 +1112,15 @@ Status AiCpuCCTask::LaunchKernel(const std::vector<GeTensorDesc> &input_desc, |
|
|
|
std::vector<DataBuffer> &output_buffers, |
|
|
|
rtStream_t 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)); |
|
|
|
if (unknown_type_ == DEPEND_SHAPE_RANGE) { |
|
|
|
GE_CHK_RT_RET(rtStreamSynchronize(stream)); |
|
|
|