wehub-resource-sync
57 lines
2.2 KiB
C++
57 lines
2.2 KiB
C++
#include <riscv_vector.h>
|
|
#if defined(__linux__) || defined(__QNXNTO__)
|
|
#include <sys/types.h>
|
|
#endif
|
|
|
|
void MNNCubicLineC16(int8_t* dst, const float* A, const float* B,
|
|
const float* C, const float* D, float* t,
|
|
int8_t* zeroPoint, size_t number,
|
|
ssize_t minValue, ssize_t maxValue) {
|
|
const float f = *t;
|
|
const float t2 = f * f, t3 = t2 * f;
|
|
const float b0 = 1.0f - 2.25f * t2 + 1.25f * t3;
|
|
const float t1 = 1.0f - f, t1_2 = t1 * t1;
|
|
const float c0 = 1.0f - 2.25f * t1_2 + 1.25f * t1_2 * t1;
|
|
const float ta = 1.0f + f, ta2 = ta * ta;
|
|
const float a0 = 3.0f - 6.0f * ta + 3.75f * ta2 - 0.75f * ta2 * ta;
|
|
const float td = 2.0f - f, td2 = td * td;
|
|
const float d0 = 3.0f - 6.0f * td + 3.75f * td2 - 0.75f * td2 * td;
|
|
const int offset = *zeroPoint;
|
|
const int minVal = (int)minValue;
|
|
const int maxVal = (int)maxValue;
|
|
const size_t total = number << 4;
|
|
size_t i = 0;
|
|
|
|
while (i < total) {
|
|
size_t vl = __riscv_vsetvl_e32m8(total - i);
|
|
vfloat32m8_t v, acc;
|
|
|
|
v = __riscv_vle32_v_f32m8(A + i, vl);
|
|
acc = __riscv_vfmul_vf_f32m8(v, a0, vl);
|
|
|
|
v = __riscv_vle32_v_f32m8(B + i, vl);
|
|
acc = __riscv_vfmacc_vf_f32m8(acc, b0, v, vl);
|
|
|
|
v = __riscv_vle32_v_f32m8(C + i, vl);
|
|
acc = __riscv_vfmacc_vf_f32m8(acc, c0, v, vl);
|
|
|
|
v = __riscv_vle32_v_f32m8(D + i, vl);
|
|
acc = __riscv_vfmacc_vf_f32m8(acc, d0, v, vl);
|
|
|
|
vfloat32m8_t half = __riscv_vfmv_v_f_f32m8(0.5f, vl);
|
|
vfloat32m8_t signHalf = __riscv_vfsgnj_vv_f32m8(half, acc, vl);
|
|
acc = __riscv_vfadd_vv_f32m8(acc, signHalf, vl);
|
|
|
|
vint32m8_t vint = __riscv_vfcvt_rtz_x_f_v_i32m8(acc, vl);
|
|
vint = __riscv_vadd_vx_i32m8(vint, offset, vl);
|
|
vint = __riscv_vmax_vx_i32m8(vint, minVal, vl);
|
|
vint = __riscv_vmin_vx_i32m8(vint, maxVal, vl);
|
|
|
|
vint16m4_t vi16 = __riscv_vncvt_x_x_w_i16m4(vint, vl);
|
|
vint8m2_t vi8 = __riscv_vncvt_x_x_w_i8m2(vi16, vl);
|
|
__riscv_vse8_v_i8m2(dst + i, vi8, vl);
|
|
|
|
i += vl;
|
|
}
|
|
}
|