根据元素数量,获取Sort数据的大小(单位为字节)。
1 2 | template <typename T> __aicore__ inline uint32_t GetSortLen(const uint32_t elemCount) |
接口 |
功能 |
---|---|
T |
操作数的数据类型。 |
接口 |
输入/输出 |
功能 |
---|---|---|
elemCount |
输入 |
输入元素个数。 |
sort数据的大小(单位为字节)。
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas推理系列产品AI Core
AscendC::GetSortLen<half>(128);