根据元素数量,获取Sort数据的大小(单位为字节)。
1 2
template <typename T> __aicore__ inline uint32_t GetSortLen(const uint32_t elemCount)
参数名
描述
T
操作数的数据类型。
Atlas A3 训练系列产品/Atlas A3 推理系列产品,支持的数据类型为:half/float
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件,支持的数据类型为:half/float
Atlas 推理系列产品AI Core,支持的数据类型为:half/float
输入/输出
elemCount
输入
输入元素个数。
sort数据的大小(单位为字节)。
无
1
AscendC::GetSortLen<half>(128);