wehub-resource-sync
38 lines
1.1 KiB
C++
38 lines
1.1 KiB
C++
#include <riscv_vector.h>
|
|
#include <cfloat>
|
|
|
|
#define UNIT 4
|
|
|
|
void MNNVectorTop1Float(float* input, float* maxValue, int32_t* maxIndex, size_t inputCountUnit) {
|
|
size_t n = inputCountUnit * UNIT;
|
|
float maxV = -FLT_MAX;
|
|
int32_t maxIdx = 0;
|
|
size_t vl;
|
|
|
|
for (size_t i = 0; i < n; ) {
|
|
vl = __riscv_vsetvl_e32m8(n - i);
|
|
vfloat32m8_t data = __riscv_vle32_v_f32m8(input + i, vl);
|
|
vfloat32m1_t scalar = __riscv_vfmv_s_f_f32m1(maxV, vl);
|
|
vfloat32m1_t result = __riscv_vfredmax_vs_f32m8_f32m1(data, scalar, vl);
|
|
maxV = __riscv_vfmv_f_s_f32m1_f32(result);
|
|
i += vl;
|
|
}
|
|
|
|
for (size_t i = 0; i < n; ) {
|
|
vl = __riscv_vsetvl_e32m8(n - i);
|
|
vfloat32m8_t data = __riscv_vle32_v_f32m8(input + i, vl);
|
|
vbool4_t mask = __riscv_vmfeq_vf_f32m8_b4(data, maxV, vl);
|
|
long first = __riscv_vfirst_m_b4(mask, vl);
|
|
|
|
if (first >= 0) {
|
|
maxIdx = i + first;
|
|
break;
|
|
}
|
|
|
|
i += vl;
|
|
}
|
|
|
|
maxValue[0] = maxV;
|
|
maxIndex[0] = maxIdx;
|
|
}
|