IterateAll产品支持情况产品是否支持Ascend 950PR/Ascend 950DTxAtlas A3 训练系列产品 / Atlas A3 推理系列产品√Atlas A2 训练系列产品 / Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品xAtlas 推理系列产品 AI CorexAtlas 推理系列产品 Vector CorexAtlas 训练系列产品x功能说明通过设置结果矩阵Output在GM上的首地址本接口一次性计算singleCo * singleDo * singleM大小的数据块并写到结果矩阵Output中。本接口提供单核内卷积计算能力singleCo为多核切分后单个核内的输出通道大小singleDo为多核切分后单个核内的Dout大小singleM为多核切分后单个核内的M大小。singleCo、singleDo和singleM的大小通过SetSingleOutputShape接口设置。函数原型__aicore__ inline void IterateAll(const AscendC::GlobalTensorOutputT output, bool enPartialSum false)参数说明参数名输入/输出描述output输入Output在GM上的地址。类型为GlobalTensor。结果矩阵Output支持的数据类型为half、bfloat16_t。enPartialSum输入预留参数。返回值说明无约束说明IterateAll接口仅支持处理单batch数据在多batch计算场景中需要通过batch次循环调用IterateAll接口完成计算。for (uint64_t batchIter 0; batchIter singleCoreBatch; batchIter) { conv3dApi.SetInput(inputGm[batchIter * inputOneBatchSize]); conv3dApi.IterateAll(outputGm[batchIter * outputOneBatchSize]); conv3dApi.End(); }IterateAll接口必须在初始化接口及输入输出配置接口之后进行调用完成Conv3D计算调用顺序如下。Init(...); ... // 输入输出配置 IterateAll(...); End();调用示例TPipe pipe; conv3dApi.Init(tiling); conv3dApi.SetWeight(weightGm); if (biasFlag) { conv3dApi.SetBias(biasGm); } conv3dApi.SetInputStartPosition(diIdxStart, mIdxStart); conv3dApi.SetSingleOutputShape(singleCoreCout, singleCoreDout, singleCoreM); for (uint64_t batchIter 0; batchIter singleCoreBatch; batchIter) { conv3dApi.SetInput(inputGm[batchIter * inputOneBatchSize]); conv3dApi.IterateAll(outputGm[batchIter * outputOneBatchSize]); conv3dApi.End(); }创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
