wehub-resource-sync
26 lines
718 B
C++
26 lines
718 B
C++
#include <riscv_vector.h>
|
|
|
|
void MNNTranspose32Bit(int32_t* dstO, const int32_t* srcO, int32_t* dim) {
|
|
int w = dim[0];
|
|
int h = dim[1];
|
|
int srcStride = dim[2];
|
|
int dstStride = dim[3];
|
|
ptrdiff_t srcStrideByte = srcStride * sizeof(int32_t);
|
|
|
|
for (int i = 0; i < h; ++i) {
|
|
const int32_t* srcPtr = srcO + i;
|
|
int32_t* dstPtr = dstO + i * dstStride;
|
|
|
|
int j = 0;
|
|
while (j < w) {
|
|
size_t vl = __riscv_vsetvl_e32m8(w - j);
|
|
vint32m8_t data = __riscv_vlse32_v_i32m8(srcPtr, srcStrideByte, vl);
|
|
__riscv_vse32_v_i32m8(dstPtr, data, vl);
|
|
srcPtr += vl * srcStride;
|
|
dstPtr += vl;
|
|
j += vl;
|
|
}
|
|
}
|
|
}
|
|
|