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