wehub-resource-sync
30 lines
1.4 KiB
C++
30 lines
1.4 KiB
C++
#include <riscv_vector.h>
|
|
|
|
void MNNAddC4WithStride(const float* source, float* dest, size_t srcStride, size_t dstStride, size_t count) {
|
|
ptrdiff_t srcStrideByte = srcStride * sizeof(float);
|
|
ptrdiff_t dstStrideByte = dstStride * sizeof(float);
|
|
size_t vl;
|
|
|
|
for (size_t i = count; i > 0; i -= vl) {
|
|
vl = __riscv_vsetvl_e32m8(i);
|
|
vfloat32m8_t vs = __riscv_vlse32_v_f32m8(source + 0, srcStrideByte, vl);
|
|
vfloat32m8_t vd = __riscv_vlse32_v_f32m8(dest + 0, dstStrideByte, vl);
|
|
vd = __riscv_vfadd_vv_f32m8(vd, vs, vl);
|
|
__riscv_vsse32_v_f32m8(dest + 0, dstStrideByte, vd, vl);
|
|
vs = __riscv_vlse32_v_f32m8(source + 1, srcStrideByte, vl);
|
|
vd = __riscv_vlse32_v_f32m8(dest + 1, dstStrideByte, vl);
|
|
vd = __riscv_vfadd_vv_f32m8(vd, vs, vl);
|
|
__riscv_vsse32_v_f32m8(dest + 1, dstStrideByte, vd, vl);
|
|
vs = __riscv_vlse32_v_f32m8(source + 2, srcStrideByte, vl);
|
|
vd = __riscv_vlse32_v_f32m8(dest + 2, dstStrideByte, vl);
|
|
vd = __riscv_vfadd_vv_f32m8(vd, vs, vl);
|
|
__riscv_vsse32_v_f32m8(dest + 2, dstStrideByte, vd, vl);
|
|
vs = __riscv_vlse32_v_f32m8(source + 3, srcStrideByte, vl);
|
|
vd = __riscv_vlse32_v_f32m8(dest + 3, dstStrideByte, vl);
|
|
vd = __riscv_vfadd_vv_f32m8(vd, vs, vl);
|
|
__riscv_vsse32_v_f32m8(dest + 3, dstStrideByte, vd, vl);
|
|
source += vl * srcStride;
|
|
dest += vl * dstStride;
|
|
}
|
|
}
|