GetReduceMaxMinCount

功能说明

获取ReduceMax、ReduceMin连续场景下的最大/最小值以及相应的索引值。

函数原型

• 获取ReduceMax、ReduceMin连续场景下的最大值与最小值，以及相应的索引值。

  1 2	template <typename T> __aicore__ inline void GetReduceMaxMinCount(T &maxMinValue, T &maxMinIndex)

• 获取ReduceMax、ReduceMin连续场景下的最大值与最小值。

  1 2	template <typename T> __aicore__ inline void GetReduceMaxMinCount(T &maxMinValue)

参数说明

返回值

ReduceMax/ReduceMin的连续场景下的最大/最小值以及相应的索引值。

支持的型号

Atlas推理系列产品AI Core，支持仅获取最值的接口，不支持获取最值和索引的接口。

Atlas A2训练系列产品/Atlas 800I A2推理产品，支持获取最值和索引的接口，不支持仅获取最值的接口。

约束说明

• 针对Atlas A2训练系列产品/Atlas 800I A2推理产品，由于ReduceMax/ReduceMin的内部实现原因，直接调用GetReduceMaxMinCount接口无法获取到准确的索引值，验证时需要使用WholeReduceMax/WholeReduceMin接口来获取准确的索引值。
• 索引maxMinIndex数据是按照ReduceMax/ReduceMin的数据类型进行存储的，比如ReduceMax/ReduceMin使用half类型时，maxMinIndex是按照half类型进行存储的，如果按照half格式进行读取，maxMinIndex的值是不对的，因此maxMinIndex的读取需要使用reinterpret_cast方法转换到整数类型，若输入数据类型是half，需要使用reinterpret_cast<uint16_t>，若输入是float，需要使用reinterpret_cast<uint32_t>。

调用示例

1. 以ReduceMax指令为例，首先执行ReduceMax指令。

   1 2 3 4 5

   AscendC::LocalTensor<float> src; AscendC::LocalTensor<float> work; AscendC::LocalTensor<float> dst; int32_t mask = 64; AscendC::ReduceMax(dst, src, work, mask, 1, 8, true); // 连续场景，srcRepStride = 8，且calIndex = true

2. 获取上述ReduceMax指令的最值与索引值。
   针对Atlas A2训练系列产品/Atlas 800I A2推理产品，需要使用WholeReduceMax指令获取准确的索引值，然后再调用GetReduceMaxMinCount指令。

   1 2 3 4 5 6 7

   AscendC::LocalTensor<float> src; AscendC::LocalTensor<float> dst; int32_t mask = 64; AscendC::WholeReduceMax(dst, src, mask, 1, 1, 1, 8); float val = 0; // 最值 float idx = 0; // 最值的索引值，与ReduceMax的结果相同 AscendC::GetReduceMaxMinCount<float>(val, idx);

   针对Atlas推理系列产品AI Core版本，则可在调用ReduceMax后直接调用GetReduceMaxMinCount指令获取其最大/最小值。

   1 2 3 4 5 6 7

   AscendC::LocalTensor<float> src; AscendC::LocalTensor<float> work; AscendC::LocalTensor<float> dst; int32_t mask = 64; AscendC::ReduceMax(dst, src, work, mask, 1, 8, true); float val = 0; // 最值 AscendC::GetReduceMaxMinCount<float>(val);