文件最后提交记录最后更新时间
Replace blockDim with numBlocks4 个月前
优化若干样例 & 新增group barrier样例4 个月前
优化若干样例 & 新增group barrier样例4 个月前
libraries && best_practices sample4 个月前
Replace blockDim with numBlocks4 个月前
updated the readme link4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
libraries && best_practices sample4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
Replace blockDim with numBlocks4 个月前
优化若干样例 & 新增group barrier样例4 个月前
README.md

归约操作算子样例介绍

概述

本样例集介绍了归约操作算子不同特性的典型用法,给出了对应的端到端实现。

算子开发样例

目录名称 功能描述
block_reduce_max 本样例基于BlockReduceMax实现最大值归约,可用于对每个datablock内所有元素求最大值
get_acc_val 本样例介绍基础api GetAccVal的调用。GetAccVal的功能是获取ReduceSum接口的计算结果
get_reduce_max_min_count 本样例介绍基础api GetReduceMaxMinCount的调用,GetReduceMaxMinCount的功能是获取ReduceMax、ReduceMin连续场景下的最大/最小值以及相应的索引值
mean 本样例介绍了基于Ascend C异构混合编程模型演示Mean融合算子的核函数直调实现,根据最后一轴的方向对各元素求平均值
pair_reduce_sum 本样例基于PairReduceSum实现求和归约,可用于相邻两个(奇数下标和偶数下标)元素求和
reduce_max_computation_of_the_first_n_data_elements 本样例基于ReduceMax的tensor前n个数据计算接口实现最大值归约,可用于指定从输入tensor的前n个数据中计算找出最大值及最大值对应的索引位置
reduce_max_high_dimensional_tensor_sharding_computation 本样例基于ReduceMax的tensor高维切分计算接口实现最大值归约,可用于从所有的输入数据中找出最大值及最大值对应的索引位置,使用mask用于控制每次迭代内参与计算的元素。
reduce_min_computation_of_the_first_n_data_elements 本样例基于ReduceMin的tensor前n个数据计算接口实现最小值归约,可用于指定从输入tensor的前n个数据中找出最小值及最小值对应的索引位置
reduce_min_high_dimensional_tensor_sharding_computation 本样例基于ReduceMin的tensor高维切分计算接口实现最小值归约,可用于从所有的输入数据中找出最小值及最小值对应的索引位置,使用mask用于控制每次迭代内参与计算的元素。
reduce_sum_computation_of_the_first_n_data_elements 本样例基于ReduceSum的tensor前n个数据计算接口实现求和归约,可用于指定对输入tensor的前n个数据求和
reduce_sum_high_dimensional_tensor_sharding_computation 本样例基于ReduceSum的tensor高维切分计算接口实现求和归约,可用于对所有的输入数据求和,使用mask用于控制每次迭代内参与计算的元素
reducemean 本样例演示了基于ReduceMean高阶API实现reducemean算子。样例对一个多维向量按照指定的维度求平均值
reducemin 本样例演示了调用ReduceMin高阶API实现reducemin算子。样例对一个多维向量在指定的维度求最小值
reducesum_lowlevel 本样例演示了基于ReduceSum高阶API实现reducesum算子。样例对一个多维向量按照指定的维度进行数据累加
repeat_reduce_sum 本样例基于RepeatReduceSum实现求和归约,可用于每个repeat内所有数据求和
sum 本样例介绍了调用Sum高阶API实现sum算子,获取输入数据最后一个维度的元素总和
whole_reduce_max 本样例基于WholeReduceMax实现最大值归约,可用于对每个repeat内所有数据求最大值以及其索引index,返回的索引值为每个repeat内部索引
whole_reduce_min 本样例基于WholeReduceMin实现获得每个repeat内所有数据的最小值及其索引index的功能,返回的索引值为每个repeat内部索引
whole_reduce_sum 本样例基于WholeReduceSum实现对每个repeat内所有数据求和的功能