optimize pad with rvv.

kendryte · Mar 3, 2025 · c0312cf · c0312cf
1 parent b2f080c
commit c0312cf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/Native/src/kernels/stackvm/reference/pad.cpp b/src/Native/src/kernels/stackvm/reference/pad.cpp
@@ -19,6 +19,7 @@
 #include <nncase/runtime/host_buffer.h>
 #include <nncase/runtime/runtime_op_utility.h>
 #include <nncase/runtime/util.h>
+#include "../optimized/opt_common.h"
 
 using namespace nncase;
 using namespace nncase::runtime;
@@ -101,9 +102,7 @@ pad_impl(const T *input, T *output, gsl::span<const size_t> in_shape,
 
 template <class T>
 void copy_data_v(T *src, T *dst, int blocks_in, int blocks_out, T value) {
-    for (int i = 0; i < blocks_in; ++i) {
-        dst[i] = src[i];
-    }
+    opt_memcpy(dst, src, blocks_in * sizeof(T));
     dst += blocks_in;
     for (int i = 0; i < blocks_out - blocks_in; ++i) {
         dst[i] = value;