diff --git a/tests/test_copyto_1.cpp b/tests/test_copyto_1.cpp index a381cdabf51..48bc3b958ed 100644 --- a/tests/test_copyto_1.cpp +++ b/tests/test_copyto_1.cpp @@ -14,58 +14,70 @@ #include "testutil.h" -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_copyto(const ncnn::Mat& self, const ncnn::Mat& src, const ncnn::Mat& starts, const ncnn::Mat& axes) +static int test_copyto(const ncnn::Mat& self, const ncnn::Mat& src, const std::vector& starts_array, const std::vector& axes_array) { + ncnn::Mat starts(starts_array.size()); + { + int* p = starts; + for (size_t i = 0; i < starts_array.size(); i++) + { + p[i] = starts_array[i]; + } + } + + ncnn::Mat axes(axes_array.size()); + { + int* p = axes; + for (size_t i = 0; i < axes_array.size(); i++) + { + p[i] = axes_array[i]; + } + } + ncnn::ParamDict pd; pd.set(9, starts); // starts pd.set(11, axes); // axes @@ -81,9 +93,9 @@ static int test_copyto(const ncnn::Mat& self, const ncnn::Mat& src, const ncnn:: { fprintf(stderr, "test_copyto failed self.dims=%d self=(%d %d %d %d) src.dims=%d src=(%d %d %d %d)", self.dims, self.w, self.h, self.d, self.c, src.dims, src.w, src.h, src.d, src.c); fprintf(stderr, " starts="); - print_int_array(starts); + print_int_array(starts_array); fprintf(stderr, " axes="); - print_int_array(axes); + print_int_array(axes_array); fprintf(stderr, "\n"); } @@ -111,10 +123,10 @@ static int test_copyto_0() const ncnn::Mat& src = b[j]; int ret = 0 - || test_copyto(self, src, IntArrayMat(0), IntArrayMat(0)) - || test_copyto(self, src, IntArrayMat(13), IntArrayMat(-1)) - || test_copyto(self, src, IntArrayMat(28), IntArrayMat(0)) - || test_copyto(self, src, IntArrayMat(32), ncnn::Mat()); + || test_copyto(self, src, IntArray(0), IntArray(0)) + || test_copyto(self, src, IntArray(13), IntArray(-1)) + || test_copyto(self, src, IntArray(28), IntArray(0)) + || test_copyto(self, src, IntArray(32), std::vector()); if (ret != 0) return ret; @@ -148,10 +160,10 @@ static int test_copyto_1() const ncnn::Mat& src = b[j]; int ret = 0 - || test_copyto(self, src, IntArrayMat(0, 0), IntArrayMat(0, 1)) - || test_copyto(self, src, IntArrayMat(13, 1), IntArrayMat(-2, -1)) - || test_copyto(self, src, IntArrayMat(28, 3), IntArrayMat(0, 1)) - || test_copyto(self, src, IntArrayMat(32, 10), IntArrayMat(0, 1)); + || test_copyto(self, src, IntArray(0, 0), IntArray(0, 1)) + || test_copyto(self, src, IntArray(13, 1), IntArray(-2, -1)) + || test_copyto(self, src, IntArray(28, 3), IntArray(0, 1)) + || test_copyto(self, src, IntArray(32, 10), IntArray(0, 1)); if (ret != 0) return ret; @@ -188,10 +200,10 @@ static int test_copyto_2() const ncnn::Mat& src = b[j]; int ret = 0 - || test_copyto(self, src, IntArrayMat(0, 0, 0), IntArrayMat(0, 1, 2)) - || test_copyto(self, src, IntArrayMat(13, 1, 0), IntArrayMat(-3, -2, -1)) - || test_copyto(self, src, IntArrayMat(28, 3, 4), IntArrayMat(0, 1, 2)) - || test_copyto(self, src, IntArrayMat(32, 0, 5), IntArrayMat(0, 1, 2)); + || test_copyto(self, src, IntArray(0, 0, 0), IntArray(0, 1, 2)) + || test_copyto(self, src, IntArray(13, 1, 0), IntArray(-3, -2, -1)) + || test_copyto(self, src, IntArray(28, 3, 4), IntArray(0, 1, 2)) + || test_copyto(self, src, IntArray(32, 0, 5), IntArray(0, 1, 2)); if (ret != 0) return ret; @@ -231,10 +243,10 @@ static int test_copyto_3() const ncnn::Mat& src = b[j]; int ret = 0 - || test_copyto(self, src, IntArrayMat(0, 0, 0, 0), IntArrayMat(0, 1, 2, 3)) - || test_copyto(self, src, IntArrayMat(13, 1, 1, 0), IntArrayMat(-4, -3, 2, 3)) - || test_copyto(self, src, IntArrayMat(28, 0, 3, 4), IntArrayMat(0, 1, 2, 3)) - || test_copyto(self, src, IntArrayMat(32, 2, 0, 5), IntArrayMat(0, 1, 2, 3)); + || test_copyto(self, src, IntArray(0, 0, 0, 0), IntArray(0, 1, 2, 3)) + || test_copyto(self, src, IntArray(13, 1, 1, 0), IntArray(-4, -3, 2, 3)) + || test_copyto(self, src, IntArray(28, 0, 3, 4), IntArray(0, 1, 2, 3)) + || test_copyto(self, src, IntArray(32, 2, 0, 5), IntArray(0, 1, 2, 3)); if (ret != 0) return ret; diff --git a/tests/test_crop_1.cpp b/tests/test_crop_1.cpp index 3064dc1de69..32c829c17df 100644 --- a/tests/test_crop_1.cpp +++ b/tests/test_crop_1.cpp @@ -14,58 +14,79 @@ #include "testutil.h" -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_crop(const ncnn::Mat& a, const ncnn::Mat& starts, const ncnn::Mat& ends, const ncnn::Mat& axes) +static int test_crop(const ncnn::Mat& a, const std::vector& starts_array, const std::vector& ends_array, const std::vector& axes_array) { + ncnn::Mat starts(starts_array.size()); + { + int* p = starts; + for (size_t i = 0; i < starts_array.size(); i++) + { + p[i] = starts_array[i]; + } + } + + ncnn::Mat ends(ends_array.size()); + { + int* p = ends; + for (size_t i = 0; i < ends_array.size(); i++) + { + p[i] = ends_array[i]; + } + } + + ncnn::Mat axes(axes_array.size()); + { + int* p = axes; + for (size_t i = 0; i < axes_array.size(); i++) + { + p[i] = axes_array[i]; + } + } + ncnn::ParamDict pd; pd.set(9, starts); // starts pd.set(10, ends); // ends @@ -78,282 +99,277 @@ static int test_crop(const ncnn::Mat& a, const ncnn::Mat& starts, const ncnn::Ma { fprintf(stderr, "test_crop failed a.dims=%d a=(%d %d %d %d)", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " starts="); - print_int_array(starts); + print_int_array(starts_array); fprintf(stderr, " ends="); - print_int_array(ends); + print_int_array(ends_array); fprintf(stderr, " axes="); - print_int_array(axes); + print_int_array(axes_array); fprintf(stderr, "\n"); } return ret; } -static int test_crop_1(const ncnn::Mat& a) +static int test_crop_1d(const ncnn::Mat& a) { - return 0 - || test_crop(a, IntArrayMat(12), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(16), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(11), IntArrayMat(11 + 16), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(12 + 7), IntArrayMat(-1)) - || test_crop(a, IntArrayMat(16), IntArrayMat(16 + 12), ncnn::Mat()) - || test_crop(a, IntArrayMat(11), IntArrayMat(-7 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(-12 + 1), IntArrayMat(-1)) - || test_crop(a, IntArrayMat(16), IntArrayMat(-16 + 1), ncnn::Mat()); + std::vector params[][3] = + { + {IntArray(12), IntArray(-233), IntArray(0)}, + {IntArray(16), IntArray(-233), IntArray(0)}, + {IntArray(11), IntArray(11 + 16), IntArray(0)}, + {IntArray(12), IntArray(12 + 7), IntArray(-1)}, + {IntArray(16), IntArray(16 + 12), std::vector()}, + {IntArray(11), IntArray(-7 + 1), IntArray(0)}, + {IntArray(12), IntArray(-12 + 1), IntArray(-1)}, + {IntArray(16), IntArray(-16 + 1), std::vector()} + }; + + for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++) + { + int ret = test_crop(a, params[i][0], params[i][1], params[i][2]); + if (ret) + return ret; + } + + return 0; } -static int test_crop_4(const ncnn::Mat& a) +static int test_crop_2d(const ncnn::Mat& a) { - return 0 - || test_crop(a, IntArrayMat(12), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-233), IntArrayMat(1)) - || test_crop(a, IntArrayMat(5, 11), IntArrayMat(-233, -233), IntArrayMat(0, 1)) - - || test_crop(a, IntArrayMat(11), IntArrayMat(11 + 16), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(12 + 7), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(8 + 12), IntArrayMat(-2)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(8), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(9), IntArrayMat(1)) - || test_crop(a, IntArrayMat(4), IntArrayMat(12), IntArrayMat(-1)) - - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(11 + 7, 11), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(12 + 12, 12), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(8 + 16, 10), IntArrayMat(0, -1)) - - || test_crop(a, IntArrayMat(11), IntArrayMat(-16 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(-7 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(-12 + 1), IntArrayMat(-2)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(-5 + 1), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-6 + 1), IntArrayMat(1)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-4 + 1), IntArrayMat(-1)) + std::vector params[][3] = + { + {IntArray(12), IntArray(-233), IntArray(0)}, + {IntArray(8), IntArray(-233), IntArray(0)}, + {IntArray(4), IntArray(-233), IntArray(1)}, + {IntArray(5, 11), IntArray(-233, -233), IntArray(0, 1)}, + {IntArray(11), IntArray(11 + 16), IntArray(0)}, + {IntArray(12), IntArray(12 + 7), IntArray(0)}, + {IntArray(8), IntArray(8 + 12), IntArray(-2)}, + {IntArray(5), IntArray(8), IntArray(1)}, + {IntArray(6), IntArray(9), IntArray(1)}, + {IntArray(4), IntArray(12), IntArray(-1)}, + {IntArray(11, 5), IntArray(11 + 7, 11), IntArray(0, 1)}, + {IntArray(12, 6), IntArray(12 + 12, 12), IntArray(0, 1)}, + {IntArray(8, 4), IntArray(8 + 16, 10), IntArray(0, -1)}, + {IntArray(11), IntArray(-16 + 1), IntArray(0)}, + {IntArray(12), IntArray(-7 + 1), IntArray(0)}, + {IntArray(8), IntArray(-12 + 1), IntArray(-2)}, + {IntArray(5), IntArray(-5 + 1), IntArray(1)}, + {IntArray(6), IntArray(-6 + 1), IntArray(1)}, + {IntArray(4), IntArray(-4 + 1), IntArray(-1)}, + {IntArray(11, 5), IntArray(-12 + 1, -6 + 1), IntArray(0, 1)}, + {IntArray(12, 6), IntArray(-16 + 1, -5 + 1), IntArray(0, 1)}, + {IntArray(8, 4), IntArray(-7 + 1, -4 + 1), IntArray(-2, -1)} + }; + + for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++) + { + int ret = test_crop(a, params[i][0], params[i][1], params[i][2]); + if (ret) + return ret; + } - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(-12 + 1, -6 + 1), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(-16 + 1, -5 + 1), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(-7 + 1, -4 + 1), IntArrayMat(-2, -1)); + return 0; } -static int test_crop_7(const ncnn::Mat& a) +static int test_crop_3d(const ncnn::Mat& a) { - return 0 - || test_crop(a, IntArrayMat(11), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(5), IntArrayMat(-233), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-233), IntArrayMat(2)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-233), IntArrayMat(-1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(-233, -233), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(-233, -233), IntArrayMat(0, -1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(-233, -233), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(6, 6), IntArrayMat(-233, -233), IntArrayMat(1, -1)) - || test_crop(a, IntArrayMat(11, 5, 5), IntArrayMat(-233, -233, -233), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(8, 4, 4), IntArrayMat(-233, -233, -233), IntArrayMat(0, 1, -1)) - - || test_crop(a, IntArrayMat(11), IntArrayMat(11 + 7), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(12 + 12), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(8 + 16), IntArrayMat(0)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(13), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(12), IntArrayMat(1)) - || test_crop(a, IntArrayMat(4), IntArrayMat(11), IntArrayMat(-2)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(12), IntArrayMat(2)) - || test_crop(a, IntArrayMat(6), IntArrayMat(11), IntArrayMat(2)) - || test_crop(a, IntArrayMat(4), IntArrayMat(13), IntArrayMat(-1)) - - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(11 + 7, 11), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(12 + 16, 12), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(8 + 12, 13), IntArrayMat(0, -2)) - - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(11 + 16, 13), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(12 + 12, 11), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(8 + 7, 12), IntArrayMat(0, -1)) - - || test_crop(a, IntArrayMat(5, 4), IntArrayMat(12, 12), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(6, 3), IntArrayMat(13, 13), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(4, 2), IntArrayMat(11, 11), IntArrayMat(-2, -1)) - - || test_crop(a, IntArrayMat(11, 5, 2), IntArrayMat(11 + 7, 11, 11), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(12, 6, 4), IntArrayMat(12 + 16, 12, 12), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(8, 4, 3), IntArrayMat(8 + 12, 13, 13), IntArrayMat(-3, -2, -1)) - - || test_crop(a, IntArrayMat(11), IntArrayMat(-7 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(-12 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(-16 + 1), IntArrayMat(-3)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(-6 + 1), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-5 + 1), IntArrayMat(1)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-4 + 1), IntArrayMat(-2)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(-5 + 1), IntArrayMat(2)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-4 + 1), IntArrayMat(2)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-6 + 1), IntArrayMat(-1)) - - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(-7 + 1, -4 + 1), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(-12 + 1, -6 + 1), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(-16 + 1, -5 + 1), IntArrayMat(-3, -2)) - - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(-12 + 1, -6 + 1), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(-16 + 1, -5 + 1), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(-7 + 1, -4 + 1), IntArrayMat(-3, -1)) - - || test_crop(a, IntArrayMat(5, 2), IntArrayMat(-5 + 1, -5 + 1), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(6, 4), IntArrayMat(-4 + 1, -4 + 1), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(4, 3), IntArrayMat(-6 + 1, -6 + 1), IntArrayMat(-2, -1)) + std::vector params[][3] = + { + {IntArray(11), IntArray(-233), IntArray(0)}, + {IntArray(8), IntArray(-233), IntArray(0)}, + {IntArray(5), IntArray(-233), IntArray(1)}, + {IntArray(6), IntArray(-233), IntArray(2)}, + {IntArray(4), IntArray(-233), IntArray(-1)}, + {IntArray(12, 6), IntArray(-233, -233), IntArray(0, 1)}, + {IntArray(11, 5), IntArray(-233, -233), IntArray(0, -1)}, + {IntArray(8, 4), IntArray(-233, -233), IntArray(0, 2)}, + {IntArray(6, 6), IntArray(-233, -233), IntArray(1, -1)}, + {IntArray(11, 5, 5), IntArray(-233, -233, -233), IntArray(0, 1, 2)}, + {IntArray(8, 4, 4), IntArray(-233, -233, -233), IntArray(0, 1, -1)}, + {IntArray(11), IntArray(11 + 7), IntArray(0)}, + {IntArray(12), IntArray(12 + 12), IntArray(0)}, + {IntArray(8), IntArray(8 + 16), IntArray(0)}, + {IntArray(5), IntArray(13), IntArray(1)}, + {IntArray(6), IntArray(12), IntArray(1)}, + {IntArray(4), IntArray(11), IntArray(-2)}, + {IntArray(5), IntArray(12), IntArray(2)}, + {IntArray(6), IntArray(11), IntArray(2)}, + {IntArray(4), IntArray(13), IntArray(-1)}, + {IntArray(11, 5), IntArray(11 + 7, 11), IntArray(0, 1)}, + {IntArray(12, 6), IntArray(12 + 16, 12), IntArray(0, 1)}, + {IntArray(8, 4), IntArray(8 + 12, 13), IntArray(0, -2)}, + {IntArray(11, 5), IntArray(11 + 16, 13), IntArray(0, 2)}, + {IntArray(12, 6), IntArray(12 + 12, 11), IntArray(0, 2)}, + {IntArray(8, 4), IntArray(8 + 7, 12), IntArray(0, -1)}, + {IntArray(5, 4), IntArray(12, 12), IntArray(1, 2)}, + {IntArray(6, 3), IntArray(13, 13), IntArray(1, 2)}, + {IntArray(4, 2), IntArray(11, 11), IntArray(-2, -1)}, + {IntArray(11, 5, 2), IntArray(11 + 7, 11, 11), IntArray(0, 1, 2)}, + {IntArray(12, 6, 4), IntArray(12 + 16, 12, 12), IntArray(0, 1, 2)}, + {IntArray(8, 4, 3), IntArray(8 + 12, 13, 13), IntArray(-3, -2, -1)}, + {IntArray(11), IntArray(-7 + 1), IntArray(0)}, + {IntArray(12), IntArray(-12 + 1), IntArray(0)}, + {IntArray(8), IntArray(-16 + 1), IntArray(-3)}, + {IntArray(5), IntArray(-6 + 1), IntArray(1)}, + {IntArray(6), IntArray(-5 + 1), IntArray(1)}, + {IntArray(4), IntArray(-4 + 1), IntArray(-2)}, + {IntArray(5), IntArray(-5 + 1), IntArray(2)}, + {IntArray(6), IntArray(-4 + 1), IntArray(2)}, + {IntArray(4), IntArray(-6 + 1), IntArray(-1)}, + {IntArray(11, 5), IntArray(-7 + 1, -4 + 1), IntArray(0, 1)}, + {IntArray(12, 6), IntArray(-12 + 1, -6 + 1), IntArray(0, 1)}, + {IntArray(8, 4), IntArray(-16 + 1, -5 + 1), IntArray(-3, -2)}, + {IntArray(11, 5), IntArray(-12 + 1, -6 + 1), IntArray(0, 2)}, + {IntArray(12, 6), IntArray(-16 + 1, -5 + 1), IntArray(0, 2)}, + {IntArray(8, 4), IntArray(-7 + 1, -4 + 1), IntArray(-3, -1)}, + {IntArray(5, 2), IntArray(-5 + 1, -5 + 1), IntArray(1, 2)}, + {IntArray(6, 4), IntArray(-4 + 1, -4 + 1), IntArray(1, 2)}, + {IntArray(4, 3), IntArray(-6 + 1, -6 + 1), IntArray(-2, -1)}, + {IntArray(11, 5, 4), IntArray(-7 + 1, -5 + 1, -5 + 1), IntArray(0, 1, 2)}, + {IntArray(12, 6, 3), IntArray(-12 + 1, -6 + 1, -6 + 1), IntArray(0, 1, 2)}, + {IntArray(8, 4, 2), IntArray(-16 + 1, -4 + 1, -4 + 1), IntArray(-3, -2, -1)} + }; + + + for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++) + { + int ret = test_crop(a, params[i][0], params[i][1], params[i][2]); + if (ret) + return ret; + } - || test_crop(a, IntArrayMat(11, 5, 4), IntArrayMat(-7 + 1, -5 + 1, -5 + 1), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(12, 6, 3), IntArrayMat(-12 + 1, -6 + 1, -6 + 1), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(8, 4, 2), IntArrayMat(-16 + 1, -4 + 1, -4 + 1), IntArrayMat(-3, -2, -1)); + return 0; } -static int test_crop_10(const ncnn::Mat& a) +static int test_crop_4d(const ncnn::Mat& a) { - return 0 - || test_crop(a, IntArrayMat(11), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(-233), IntArrayMat(0)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-233), IntArrayMat(1)) - || test_crop(a, IntArrayMat(5), IntArrayMat(-233), IntArrayMat(2)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-233), IntArrayMat(-2)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-233), IntArrayMat(3)) - || test_crop(a, IntArrayMat(5), IntArrayMat(-233), IntArrayMat(-1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(-233, -233), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(-233, -233), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(-233, -233), IntArrayMat(-4, -2)) - || test_crop(a, IntArrayMat(4, 4), IntArrayMat(-233, -233), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(-233, -233), IntArrayMat(0, 3)) - || test_crop(a, IntArrayMat(5, 5), IntArrayMat(-233, -233), IntArrayMat(1, 3)) - || test_crop(a, IntArrayMat(4, 4), IntArrayMat(-233, -233), IntArrayMat(2, 3)) - || test_crop(a, IntArrayMat(12, 6, 6), IntArrayMat(-233, -233, -233), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(11, 5, 5), IntArrayMat(-233, -233, -233), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(8, 4, 4), IntArrayMat(-233, -233, -233), IntArrayMat(0, 1, 3)) - || test_crop(a, IntArrayMat(12, 6, 6), IntArrayMat(-233, -233, -233), IntArrayMat(0, 2, 3)) - || test_crop(a, IntArrayMat(11, 5, 5), IntArrayMat(-233, -233, -233), IntArrayMat(0, 2, 3)) - || test_crop(a, IntArrayMat(4, 4, 4), IntArrayMat(-233, -233, -233), IntArrayMat(1, 2, 3)) - || test_crop(a, IntArrayMat(6, 6, 6), IntArrayMat(-233, -233, -233), IntArrayMat(1, 2, 3)) - || test_crop(a, IntArrayMat(11, 5, 5, 5), IntArrayMat(-233, -233, -233, -233), IntArrayMat(0, 1, 2, 3)) - || test_crop(a, IntArrayMat(8, 4, 4, 4), IntArrayMat(-233, -233, -233, -233), IntArrayMat(0, 1, 2, 3)) - || test_crop(a, IntArrayMat(12, 6, 6, 6), IntArrayMat(-233, -233, -233, -233), IntArrayMat(-4, -3, -2, -1)) - - || test_crop(a, IntArrayMat(11), IntArrayMat(11 + 16), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(12 + 7), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(8 + 12), IntArrayMat(-4)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(11), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(13), IntArrayMat(1)) - || test_crop(a, IntArrayMat(4), IntArrayMat(12), IntArrayMat(-3)) - - || test_crop(a, IntArrayMat(3), IntArrayMat(12), IntArrayMat(2)) - || test_crop(a, IntArrayMat(4), IntArrayMat(13), IntArrayMat(2)) - || test_crop(a, IntArrayMat(5), IntArrayMat(11), IntArrayMat(-2)) - - || test_crop(a, IntArrayMat(1), IntArrayMat(8), IntArrayMat(3)) - || test_crop(a, IntArrayMat(2), IntArrayMat(7), IntArrayMat(3)) - || test_crop(a, IntArrayMat(3), IntArrayMat(6), IntArrayMat(-1)) - - || test_crop(a, IntArrayMat(11, 5), IntArrayMat(11 + 7, 11), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 6), IntArrayMat(12 + 12, 12), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(8, 4), IntArrayMat(8 + 16, 13), IntArrayMat(-4, -3)) - - || test_crop(a, IntArrayMat(11, 4), IntArrayMat(11 + 12, 13), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(12, 3), IntArrayMat(12 + 16, 11), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(8, 2), IntArrayMat(8 + 7, 12), IntArrayMat(-4, -2)) - - || test_crop(a, IntArrayMat(11, 1), IntArrayMat(11 + 16, 5), IntArrayMat(0, 3)) - || test_crop(a, IntArrayMat(12, 2), IntArrayMat(12 + 7, 6), IntArrayMat(0, 3)) - || test_crop(a, IntArrayMat(8, 3), IntArrayMat(8 + 12, 7), IntArrayMat(-4, -1)) - - || test_crop(a, IntArrayMat(3, 3), IntArrayMat(13, 4), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(4, 2), IntArrayMat(12, 3), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(5, 1), IntArrayMat(11, 2), IntArrayMat(-3, -2)) - - || test_crop(a, IntArrayMat(5, 5), IntArrayMat(11, 8), IntArrayMat(1, 3)) - || test_crop(a, IntArrayMat(4, 6), IntArrayMat(12, 9), IntArrayMat(1, 3)) - || test_crop(a, IntArrayMat(3, 4), IntArrayMat(13, 7), IntArrayMat(-3, -1)) - - || test_crop(a, IntArrayMat(2, 3), IntArrayMat(12, 9), IntArrayMat(2, 3)) - || test_crop(a, IntArrayMat(3, 2), IntArrayMat(11, 7), IntArrayMat(2, 3)) - || test_crop(a, IntArrayMat(4, 1), IntArrayMat(10, 8), IntArrayMat(-2, -1)) - - || test_crop(a, IntArrayMat(11, 2, 2), IntArrayMat(11 + 6, 9, 9), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(12, 3, 3), IntArrayMat(12 + 1, 10, 10), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(8, 4, 4), IntArrayMat(8 + 3, 11, 11), IntArrayMat(-4, -3, -2)) - - || test_crop(a, IntArrayMat(11, 4, 4), IntArrayMat(11 + 12, 12, 12), IntArrayMat(0, 1, 3)) - || test_crop(a, IntArrayMat(12, 5, 5), IntArrayMat(12 + 8, 11, 11), IntArrayMat(0, 1, 3)) - || test_crop(a, IntArrayMat(8, 6, 6), IntArrayMat(8 + 4, 13, 13), IntArrayMat(-4, -3, -1)) - - || test_crop(a, IntArrayMat(11, 1, 4), IntArrayMat(11 + 5, 12, 12), IntArrayMat(0, 2, 3)) - || test_crop(a, IntArrayMat(12, 3, 3), IntArrayMat(12 + 3, 11, 11), IntArrayMat(0, 2, 3)) - || test_crop(a, IntArrayMat(8, 2, 5), IntArrayMat(8 + 2, 10, 10), IntArrayMat(-4, -2, -1)) - - || test_crop(a, IntArrayMat(1, 1, 1), IntArrayMat(7, 7, 7), IntArrayMat(1, 2, 3)) - || test_crop(a, IntArrayMat(2, 2, 2), IntArrayMat(8, 9, 10), IntArrayMat(1, 2, 3)) - || test_crop(a, IntArrayMat(3, 3, 3), IntArrayMat(11, 12, 13), IntArrayMat(-3, -2, -1)) - - || test_crop(a, IntArrayMat(11, 2, 3, 6), IntArrayMat(11 + 11, 10, 12, 11), IntArrayMat(0, 1, 2, 3)) - || test_crop(a, IntArrayMat(12, 3, 4, 5), IntArrayMat(12 + 12, 9, 11, 13), IntArrayMat(0, 1, 2, 3)) - || test_crop(a, IntArrayMat(8, 4, 5, 4), IntArrayMat(8 + 8, 8, 10, 12), IntArrayMat(-4, -3, -2, -1)) - - || test_crop(a, IntArrayMat(11), IntArrayMat(-7 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(12), IntArrayMat(-12 + 1), IntArrayMat(0)) - || test_crop(a, IntArrayMat(8), IntArrayMat(-16 + 1), IntArrayMat(-4)) - - || test_crop(a, IntArrayMat(5), IntArrayMat(-6 + 1), IntArrayMat(1)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-5 + 1), IntArrayMat(1)) - || test_crop(a, IntArrayMat(4), IntArrayMat(-4 + 1), IntArrayMat(-3)) - - || test_crop(a, IntArrayMat(4), IntArrayMat(-4 + 1), IntArrayMat(2)) - || test_crop(a, IntArrayMat(5), IntArrayMat(-5 + 1), IntArrayMat(2)) - || test_crop(a, IntArrayMat(6), IntArrayMat(-6 + 1), IntArrayMat(-2)) - - || test_crop(a, IntArrayMat(1), IntArrayMat(-5 + 1), IntArrayMat(3)) - || test_crop(a, IntArrayMat(2), IntArrayMat(-4 + 1), IntArrayMat(3)) - || test_crop(a, IntArrayMat(3), IntArrayMat(-3 + 1), IntArrayMat(-1)) - - || test_crop(a, IntArrayMat(11, 3), IntArrayMat(-7 + 1, -3 + 1), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(12, 4), IntArrayMat(-12 + 1, -4 + 1), IntArrayMat(0, 1)) - || test_crop(a, IntArrayMat(8, 5), IntArrayMat(-16 + 1, -5 + 1), IntArrayMat(-4, -3)) - - || test_crop(a, IntArrayMat(11, 1), IntArrayMat(-12 + 1, -5 + 1), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(12, 2), IntArrayMat(-16 + 1, -4 + 1), IntArrayMat(0, 2)) - || test_crop(a, IntArrayMat(8, 3), IntArrayMat(-7 + 1, -6 + 1), IntArrayMat(-4, -2)) - - || test_crop(a, IntArrayMat(11, 3), IntArrayMat(-12 + 1, -2 + 1), IntArrayMat(0, 3)) - || test_crop(a, IntArrayMat(12, 4), IntArrayMat(-16 + 1, -3 + 1), IntArrayMat(0, 3)) - || test_crop(a, IntArrayMat(8, 5), IntArrayMat(-7 + 1, -4 + 1), IntArrayMat(-4, -1)) - - || test_crop(a, IntArrayMat(2, 3), IntArrayMat(-4 + 1, -2 + 1), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(3, 4), IntArrayMat(-2 + 1, -3 + 1), IntArrayMat(1, 2)) - || test_crop(a, IntArrayMat(4, 5), IntArrayMat(-3 + 1, -4 + 1), IntArrayMat(-3, -2)) - - || test_crop(a, IntArrayMat(3, 2), IntArrayMat(-2 + 1, -4 + 1), IntArrayMat(1, 3)) - || test_crop(a, IntArrayMat(4, 3), IntArrayMat(-3 + 1, -2 + 1), IntArrayMat(1, 3)) - || test_crop(a, IntArrayMat(5, 4), IntArrayMat(-4 + 1, -3 + 1), IntArrayMat(-3, -1)) - - || test_crop(a, IntArrayMat(2, 3), IntArrayMat(-4 + 1, -6 + 1), IntArrayMat(2, 3)) - || test_crop(a, IntArrayMat(1, 2), IntArrayMat(-5 + 1, -5 + 1), IntArrayMat(2, 3)) - || test_crop(a, IntArrayMat(3, 1), IntArrayMat(-6 + 1, -4 + 1), IntArrayMat(-2, -1)) - - || test_crop(a, IntArrayMat(11, 3, 3), IntArrayMat(-7 + 1, -3 + 1, -4 + 1), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(12, 4, 4), IntArrayMat(-12 + 1, -4 + 1, -3 + 1), IntArrayMat(0, 1, 2)) - || test_crop(a, IntArrayMat(8, 5, 5), IntArrayMat(-16 + 1, -5 + 1, -5 + 1), IntArrayMat(-4, -3, -2)) - - || test_crop(a, IntArrayMat(11, 2, 2), IntArrayMat(-7 + 1, -5 + 1, -4 + 1), IntArrayMat(0, 1, 3)) - || test_crop(a, IntArrayMat(12, 1, 1), IntArrayMat(-12 + 1, -6 + 1, -5 + 1), IntArrayMat(0, 1, 3)) - || test_crop(a, IntArrayMat(8, 3, 3), IntArrayMat(-16 + 1, -4 + 1, -6 + 1), IntArrayMat(-4, -3, -1)) - - || test_crop(a, IntArrayMat(11, 2, 5), IntArrayMat(-7 + 1, -2 + 1, -5 + 1), IntArrayMat(0, 2, 3)) - || test_crop(a, IntArrayMat(12, 3, 3), IntArrayMat(-12 + 1, -3 + 1, -4 + 1), IntArrayMat(0, 2, 3)) - || test_crop(a, IntArrayMat(8, 4, 4), IntArrayMat(-16 + 1, -4 + 1, -3 + 1), IntArrayMat(-4, -2, -1)) - - || test_crop(a, IntArrayMat(1, 3, 3), IntArrayMat(-3 + 1, -6 + 1, -4 + 1), IntArrayMat(1, 2, 3)) - || test_crop(a, IntArrayMat(2, 2, 2), IntArrayMat(-4 + 1, -4 + 1, -5 + 1), IntArrayMat(1, 2, 3)) - || test_crop(a, IntArrayMat(3, 1, 1), IntArrayMat(-5 + 1, -5 + 1, -6 + 1), IntArrayMat(-3, -2, -1)) + std::vector params[][3] = + { + {IntArray(11), IntArray(-233), IntArray(0)}, + {IntArray(8), IntArray(-233), IntArray(0)}, + {IntArray(6), IntArray(-233), IntArray(1)}, + {IntArray(5), IntArray(-233), IntArray(2)}, + {IntArray(4), IntArray(-233), IntArray(-2)}, + {IntArray(6), IntArray(-233), IntArray(3)}, + {IntArray(5), IntArray(-233), IntArray(-1)}, + {IntArray(8, 4), IntArray(-233, -233), IntArray(0, 1)}, + {IntArray(12, 6), IntArray(-233, -233), IntArray(0, 2)}, + {IntArray(11, 5), IntArray(-233, -233), IntArray(-4, -2)}, + {IntArray(4, 4), IntArray(-233, -233), IntArray(1, 2)}, + {IntArray(12, 6), IntArray(-233, -233), IntArray(0, 3)}, + {IntArray(5, 5), IntArray(-233, -233), IntArray(1, 3)}, + {IntArray(4, 4), IntArray(-233, -233), IntArray(2, 3)}, + {IntArray(12, 6, 6), IntArray(-233, -233, -233), IntArray(0, 1, 2)}, + {IntArray(11, 5, 5), IntArray(-233, -233, -233), IntArray(0, 1, 2)}, + {IntArray(8, 4, 4), IntArray(-233, -233, -233), IntArray(0, 1, 3)}, + {IntArray(12, 6, 6), IntArray(-233, -233, -233), IntArray(0, 2, 3)}, + {IntArray(11, 5, 5), IntArray(-233, -233, -233), IntArray(0, 2, 3)}, + {IntArray(4, 4, 4), IntArray(-233, -233, -233), IntArray(1, 2, 3)}, + {IntArray(6, 6, 6), IntArray(-233, -233, -233), IntArray(1, 2, 3)}, + {IntArray(11, 5, 5, 5), IntArray(-233, -233, -233, -233), IntArray(0, 1, 2, 3)}, + {IntArray(8, 4, 4, 4), IntArray(-233, -233, -233, -233), IntArray(0, 1, 2, 3)}, + {IntArray(12, 6, 6, 6), IntArray(-233, -233, -233, -233), IntArray(-4, -3, -2, -1)}, + {IntArray(11), IntArray(11 + 16), IntArray(0)}, + {IntArray(12), IntArray(12 + 7), IntArray(0)}, + {IntArray(8), IntArray(8 + 12), IntArray(-4)}, + {IntArray(5), IntArray(11), IntArray(1)}, + {IntArray(6), IntArray(13), IntArray(1)}, + {IntArray(4), IntArray(12), IntArray(-3)}, + {IntArray(3), IntArray(12), IntArray(2)}, + {IntArray(4), IntArray(13), IntArray(2)}, + {IntArray(5), IntArray(11), IntArray(-2)}, + {IntArray(1), IntArray(8), IntArray(3)}, + {IntArray(2), IntArray(7), IntArray(3)}, + {IntArray(3), IntArray(6), IntArray(-1)}, + {IntArray(11, 5), IntArray(11 + 7, 11), IntArray(0, 1)}, + {IntArray(12, 6), IntArray(12 + 12, 12), IntArray(0, 1)}, + {IntArray(8, 4), IntArray(8 + 16, 13), IntArray(-4, -3)}, + {IntArray(11, 4), IntArray(11 + 12, 13), IntArray(0, 2)}, + {IntArray(12, 3), IntArray(12 + 16, 11), IntArray(0, 2)}, + {IntArray(8, 2), IntArray(8 + 7, 12), IntArray(-4, -2)}, + {IntArray(11, 1), IntArray(11 + 16, 5), IntArray(0, 3)}, + {IntArray(12, 2), IntArray(12 + 7, 6), IntArray(0, 3)}, + {IntArray(8, 3), IntArray(8 + 12, 7), IntArray(-4, -1)}, + {IntArray(3, 3), IntArray(13, 4), IntArray(1, 2)}, + {IntArray(4, 2), IntArray(12, 3), IntArray(1, 2)}, + {IntArray(5, 1), IntArray(11, 2), IntArray(-3, -2)}, + {IntArray(5, 5), IntArray(11, 8), IntArray(1, 3)}, + {IntArray(4, 6), IntArray(12, 9), IntArray(1, 3)}, + {IntArray(3, 4), IntArray(13, 7), IntArray(-3, -1)}, + {IntArray(2, 3), IntArray(12, 9), IntArray(2, 3)}, + {IntArray(3, 2), IntArray(11, 7), IntArray(2, 3)}, + {IntArray(4, 1), IntArray(10, 8), IntArray(-2, -1)}, + {IntArray(11, 2, 2), IntArray(11 + 6, 9, 9), IntArray(0, 1, 2)}, + {IntArray(12, 3, 3), IntArray(12 + 1, 10, 10), IntArray(0, 1, 2)}, + {IntArray(8, 4, 4), IntArray(8 + 3, 11, 11), IntArray(-4, -3, -2)}, + {IntArray(11, 4, 4), IntArray(11 + 12, 12, 12), IntArray(0, 1, 3)}, + {IntArray(12, 5, 5), IntArray(12 + 8, 11, 11), IntArray(0, 1, 3)}, + {IntArray(8, 6, 6), IntArray(8 + 4, 13, 13), IntArray(-4, -3, -1)}, + {IntArray(11, 1, 4), IntArray(11 + 5, 12, 12), IntArray(0, 2, 3)}, + {IntArray(12, 3, 3), IntArray(12 + 3, 11, 11), IntArray(0, 2, 3)}, + {IntArray(8, 2, 5), IntArray(8 + 2, 10, 10), IntArray(-4, -2, -1)}, + {IntArray(1, 1, 1), IntArray(7, 7, 7), IntArray(1, 2, 3)}, + {IntArray(2, 2, 2), IntArray(8, 9, 10), IntArray(1, 2, 3)}, + {IntArray(3, 3, 3), IntArray(11, 12, 13), IntArray(-3, -2, -1)}, + {IntArray(11, 2, 3, 6), IntArray(11 + 11, 10, 12, 11), IntArray(0, 1, 2, 3)}, + {IntArray(12, 3, 4, 5), IntArray(12 + 12, 9, 11, 13), IntArray(0, 1, 2, 3)}, + {IntArray(8, 4, 5, 4), IntArray(8 + 8, 8, 10, 12), IntArray(-4, -3, -2, -1)}, + {IntArray(11), IntArray(-7 + 1), IntArray(0)}, + {IntArray(12), IntArray(-12 + 1), IntArray(0)}, + {IntArray(8), IntArray(-16 + 1), IntArray(-4)}, + {IntArray(5), IntArray(-6 + 1), IntArray(1)}, + {IntArray(6), IntArray(-5 + 1), IntArray(1)}, + {IntArray(4), IntArray(-4 + 1), IntArray(-3)}, + {IntArray(4), IntArray(-4 + 1), IntArray(2)}, + {IntArray(5), IntArray(-5 + 1), IntArray(2)}, + {IntArray(6), IntArray(-6 + 1), IntArray(-2)}, + {IntArray(1), IntArray(-5 + 1), IntArray(3)}, + {IntArray(2), IntArray(-4 + 1), IntArray(3)}, + {IntArray(3), IntArray(-3 + 1), IntArray(-1)}, + {IntArray(11, 3), IntArray(-7 + 1, -3 + 1), IntArray(0, 1)}, + {IntArray(12, 4), IntArray(-12 + 1, -4 + 1), IntArray(0, 1)}, + {IntArray(8, 5), IntArray(-16 + 1, -5 + 1), IntArray(-4, -3)}, + {IntArray(11, 1), IntArray(-12 + 1, -5 + 1), IntArray(0, 2)}, + {IntArray(12, 2), IntArray(-16 + 1, -4 + 1), IntArray(0, 2)}, + {IntArray(8, 3), IntArray(-7 + 1, -6 + 1), IntArray(-4, -2)}, + {IntArray(11, 3), IntArray(-12 + 1, -2 + 1), IntArray(0, 3)}, + {IntArray(12, 4), IntArray(-16 + 1, -3 + 1), IntArray(0, 3)}, + {IntArray(8, 5), IntArray(-7 + 1, -4 + 1), IntArray(-4, -1)}, + {IntArray(2, 3), IntArray(-4 + 1, -2 + 1), IntArray(1, 2)}, + {IntArray(3, 4), IntArray(-2 + 1, -3 + 1), IntArray(1, 2)}, + {IntArray(4, 5), IntArray(-3 + 1, -4 + 1), IntArray(-3, -2)}, + {IntArray(3, 2), IntArray(-2 + 1, -4 + 1), IntArray(1, 3)}, + {IntArray(4, 3), IntArray(-3 + 1, -2 + 1), IntArray(1, 3)}, + {IntArray(5, 4), IntArray(-4 + 1, -3 + 1), IntArray(-3, -1)}, + {IntArray(2, 3), IntArray(-4 + 1, -6 + 1), IntArray(2, 3)}, + {IntArray(1, 2), IntArray(-5 + 1, -5 + 1), IntArray(2, 3)}, + {IntArray(3, 1), IntArray(-6 + 1, -4 + 1), IntArray(-2, -1)}, + {IntArray(11, 3, 3), IntArray(-7 + 1, -3 + 1, -4 + 1), IntArray(0, 1, 2)}, + {IntArray(12, 4, 4), IntArray(-12 + 1, -4 + 1, -3 + 1), IntArray(0, 1, 2)}, + {IntArray(8, 5, 5), IntArray(-16 + 1, -5 + 1, -5 + 1), IntArray(-4, -3, -2)}, + {IntArray(11, 2, 2), IntArray(-7 + 1, -5 + 1, -4 + 1), IntArray(0, 1, 3)}, + {IntArray(12, 1, 1), IntArray(-12 + 1, -6 + 1, -5 + 1), IntArray(0, 1, 3)}, + {IntArray(8, 3, 3), IntArray(-16 + 1, -4 + 1, -6 + 1), IntArray(-4, -3, -1)}, + {IntArray(11, 2, 5), IntArray(-7 + 1, -2 + 1, -5 + 1), IntArray(0, 2, 3)}, + {IntArray(12, 3, 3), IntArray(-12 + 1, -3 + 1, -4 + 1), IntArray(0, 2, 3)}, + {IntArray(8, 4, 4), IntArray(-16 + 1, -4 + 1, -3 + 1), IntArray(-4, -2, -1)}, + {IntArray(1, 3, 3), IntArray(-3 + 1, -6 + 1, -4 + 1), IntArray(1, 2, 3)}, + {IntArray(2, 2, 2), IntArray(-4 + 1, -4 + 1, -5 + 1), IntArray(1, 2, 3)}, + {IntArray(3, 1, 1), IntArray(-5 + 1, -5 + 1, -6 + 1), IntArray(-3, -2, -1)}, + {IntArray(11, 3, 4, 4), IntArray(-7 + 1, -3 + 1, -2 + 1, -4 + 1), IntArray(0, 1, 2, 3)}, + {IntArray(12, 4, 5, 3), IntArray(-12 + 1, -4 + 1, -3 + 1, -5 + 1), IntArray(0, 1, 2, 3)}, + {IntArray(8, 5, 6, 2), IntArray(-16 + 1, -5 + 1, -4 + 1, -3 + 1), IntArray(-4, -3, -2, -1)} + }; + + for (int i = 0; i < sizeof(params) / sizeof(params[0]); i++) + { + int ret = test_crop(a, params[i][0], params[i][1], params[i][2]); + if (ret) + return ret; + } - || test_crop(a, IntArrayMat(11, 3, 4, 4), IntArrayMat(-7 + 1, -3 + 1, -2 + 1, -4 + 1), IntArrayMat(0, 1, 2, 3)) - || test_crop(a, IntArrayMat(12, 4, 5, 3), IntArrayMat(-12 + 1, -4 + 1, -3 + 1, -5 + 1), IntArrayMat(0, 1, 2, 3)) - || test_crop(a, IntArrayMat(8, 5, 6, 2), IntArrayMat(-16 + 1, -5 + 1, -4 + 1, -3 + 1), IntArrayMat(-4, -3, -2, -1)); + return 0; } int main() @@ -361,16 +377,16 @@ int main() SRAND(776757); return 0 - || test_crop_1(RandomMat(112)) - || test_crop_1(RandomMat(126)) - || test_crop_1(RandomMat(127)) - || test_crop_4(RandomMat(20, 48)) - || test_crop_4(RandomMat(15, 36)) - || test_crop_4(RandomMat(16, 33)) - || test_crop_7(RandomMat(20, 20, 48)) - || test_crop_7(RandomMat(15, 15, 36)) - || test_crop_7(RandomMat(16, 16, 33)) - || test_crop_10(RandomMat(20, 20, 20, 48)) - || test_crop_10(RandomMat(15, 15, 15, 36)) - || test_crop_10(RandomMat(16, 16, 16, 33)); + || test_crop_1d(RandomMat(112)) + || test_crop_1d(RandomMat(126)) + || test_crop_1d(RandomMat(127)) + || test_crop_2d(RandomMat(20, 48)) + || test_crop_2d(RandomMat(15, 36)) + || test_crop_2d(RandomMat(16, 33)) + || test_crop_3d(RandomMat(20, 20, 48)) + || test_crop_3d(RandomMat(15, 15, 36)) + || test_crop_3d(RandomMat(16, 16, 33)) + || test_crop_4d(RandomMat(20, 20, 20, 48)) + || test_crop_4d(RandomMat(15, 15, 15, 36)) + || test_crop_4d(RandomMat(16, 16, 16, 33)); } diff --git a/tests/test_expanddims.cpp b/tests/test_expanddims.cpp index 129f9f261b1..428656282c4 100644 --- a/tests/test_expanddims.cpp +++ b/tests/test_expanddims.cpp @@ -33,58 +33,61 @@ static int test_expanddims(const ncnn::Mat& a, int expand_w, int expand_h, int e return ret; } -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_expanddims_axes(const ncnn::Mat& a, const ncnn::Mat& axes) +static int test_expanddims_axes(const ncnn::Mat& a, const std::vector& axes_array) { + ncnn::Mat axes(axes_array.size()); + { + int* p = axes; + for (size_t i = 0; i < axes_array.size(); i++) + { + p[i] = axes_array[i]; + } + } + ncnn::ParamDict pd; pd.set(3, axes); @@ -95,7 +98,7 @@ static int test_expanddims_axes(const ncnn::Mat& a, const ncnn::Mat& axes) { fprintf(stderr, "test_expanddims_axes failed a.dims=%d a=(%d %d %d %d)\n", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " axes="); - print_int_array(axes); + print_int_array(axes_array); fprintf(stderr, "\n"); } @@ -122,21 +125,21 @@ static int test_expanddims_all_params(const ncnn::Mat& a) || test_expanddims(a, 1, 1, 1, 0) || test_expanddims(a, 1, 1, 1, 1) - || test_expanddims_axes(a, IntArrayMat(0)) - || test_expanddims_axes(a, IntArrayMat(1)) - || test_expanddims_axes(a, IntArrayMat(2)) - || test_expanddims_axes(a, IntArrayMat(3)) - || test_expanddims_axes(a, IntArrayMat(0, 1)) - || test_expanddims_axes(a, IntArrayMat(0, 2)) - || test_expanddims_axes(a, IntArrayMat(0, 3)) - || test_expanddims_axes(a, IntArrayMat(1, 2)) - || test_expanddims_axes(a, IntArrayMat(1, 3)) - || test_expanddims_axes(a, IntArrayMat(2, 3)) - || test_expanddims_axes(a, IntArrayMat(0, 1, 2)) - || test_expanddims_axes(a, IntArrayMat(0, 1, 3)) - || test_expanddims_axes(a, IntArrayMat(0, 2, 3)) - || test_expanddims_axes(a, IntArrayMat(1, 2, 3)) - || test_expanddims_axes(a, IntArrayMat(0, 1, 2, 3)); + || test_expanddims_axes(a, IntArray(0)) + || test_expanddims_axes(a, IntArray(1)) + || test_expanddims_axes(a, IntArray(2)) + || test_expanddims_axes(a, IntArray(3)) + || test_expanddims_axes(a, IntArray(0, 1)) + || test_expanddims_axes(a, IntArray(0, 2)) + || test_expanddims_axes(a, IntArray(0, 3)) + || test_expanddims_axes(a, IntArray(1, 2)) + || test_expanddims_axes(a, IntArray(1, 3)) + || test_expanddims_axes(a, IntArray(2, 3)) + || test_expanddims_axes(a, IntArray(0, 1, 2)) + || test_expanddims_axes(a, IntArray(0, 1, 3)) + || test_expanddims_axes(a, IntArray(0, 2, 3)) + || test_expanddims_axes(a, IntArray(1, 2, 3)) + || test_expanddims_axes(a, IntArray(0, 1, 2, 3)); } static int test_expanddims_0() diff --git a/tests/test_slice.cpp b/tests/test_slice.cpp index dd7c8d0e23b..bbe911359e3 100644 --- a/tests/test_slice.cpp +++ b/tests/test_slice.cpp @@ -14,58 +14,61 @@ #include "testutil.h" -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_slice(const ncnn::Mat& a, const ncnn::Mat& slices, int axis) +static int test_slice(const ncnn::Mat& a, const std::vector& slices_array, int axis) { + ncnn::Mat slices(slices_array.size()); + { + int* p = slices; + for (size_t i = 0; i < slices_array.size(); i++) + { + p[i] = slices_array[i]; + } + } + ncnn::ParamDict pd; pd.set(0, slices); pd.set(1, axis); @@ -80,15 +83,24 @@ static int test_slice(const ncnn::Mat& a, const ncnn::Mat& slices, int axis) { fprintf(stderr, "test_slice failed a.dims=%d a=(%d %d %d %d)", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " slices="); - print_int_array(slices); + print_int_array(slices_array); fprintf(stderr, " axis=%d\n", axis); } return ret; } -static int test_slice_indices(const ncnn::Mat& a, const ncnn::Mat& indices, int axis) +static int test_slice_indices(const ncnn::Mat& a, const std::vector& indices_array, int axis) { + ncnn::Mat indices(indices_array.size()); + { + int* p = indices; + for (size_t i = 0; i < indices_array.size(); i++) + { + p[i] = indices_array[i]; + } + } + ncnn::ParamDict pd; pd.set(1, axis); pd.set(2, indices); @@ -103,7 +115,7 @@ static int test_slice_indices(const ncnn::Mat& a, const ncnn::Mat& indices, int { fprintf(stderr, "test_slice_indices failed a.dims=%d a=(%d %d %d %d)", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " indices="); - print_int_array(indices); + print_int_array(indices_array); fprintf(stderr, " axis=%d\n", axis); } @@ -121,20 +133,20 @@ static int test_slice_0() for (int i = 0; i < sizeof(a) / sizeof(a[0]); i++) { int ret = 0 - || test_slice(a[i], IntArrayMat(-233, -233, -233), 0) - || test_slice(a[i], IntArrayMat(-233, -233, -233), 1) - || test_slice(a[i], IntArrayMat(-233, -233, -233), -2) - || test_slice(a[i], IntArrayMat(-233, -233, -233), 3) - || test_slice(a[i], IntArrayMat(3, 12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(32, 8, -233), 0) - || test_slice(a[i], IntArrayMat(2, 12, 16, -233), 1) - || test_slice(a[i], IntArrayMat(16, 4, 5, -233), -2) - || test_slice(a[i], IntArrayMat(8, 2, 16, -233), 3) - || test_slice_indices(a[i], IntArrayMat(2, -24, -8), 0) - || test_slice_indices(a[i], IntArrayMat(4, 20, 4), 1) - || test_slice_indices(a[i], IntArrayMat(16, -16), -2) - || test_slice_indices(a[i], IntArrayMat(1, -12), 3); + || test_slice(a[i], IntArray(-233, -233, -233), 0) + || test_slice(a[i], IntArray(-233, -233, -233), 1) + || test_slice(a[i], IntArray(-233, -233, -233), -2) + || test_slice(a[i], IntArray(-233, -233, -233), 3) + || test_slice(a[i], IntArray(3, 12, 16, -233), 0) + || test_slice(a[i], IntArray(12, 16, -233), 0) + || test_slice(a[i], IntArray(32, 8, -233), 0) + || test_slice(a[i], IntArray(2, 12, 16, -233), 1) + || test_slice(a[i], IntArray(16, 4, 5, -233), -2) + || test_slice(a[i], IntArray(8, 2, 16, -233), 3) + || test_slice_indices(a[i], IntArray(2, -24, -8), 0) + || test_slice_indices(a[i], IntArray(4, 20, 4), 1) + || test_slice_indices(a[i], IntArray(16, -16), -2) + || test_slice_indices(a[i], IntArray(1, -12), 3); if (ret != 0) return ret; @@ -154,17 +166,17 @@ static int test_slice_1() for (int i = 0; i < sizeof(a) / sizeof(a[0]); i++) { int ret = 0 - || test_slice(a[i], IntArrayMat(-233, -233, -233), 0) - || test_slice(a[i], IntArrayMat(-233, -233, -233), 1) - || test_slice(a[i], IntArrayMat(-233, -233, -233), -1) - || test_slice(a[i], IntArrayMat(3, 12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(32, 8, -233), 0) - || test_slice(a[i], IntArrayMat(2, 12, 16, -233), 1) - || test_slice(a[i], IntArrayMat(16, 4, 5, -233), -1) - || test_slice_indices(a[i], IntArrayMat(2, -24, -8), 0) - || test_slice_indices(a[i], IntArrayMat(4, 20, 4), 1) - || test_slice_indices(a[i], IntArrayMat(1, -12), 2); + || test_slice(a[i], IntArray(-233, -233, -233), 0) + || test_slice(a[i], IntArray(-233, -233, -233), 1) + || test_slice(a[i], IntArray(-233, -233, -233), -1) + || test_slice(a[i], IntArray(3, 12, 16, -233), 0) + || test_slice(a[i], IntArray(12, 16, -233), 0) + || test_slice(a[i], IntArray(32, 8, -233), 0) + || test_slice(a[i], IntArray(2, 12, 16, -233), 1) + || test_slice(a[i], IntArray(16, 4, 5, -233), -1) + || test_slice_indices(a[i], IntArray(2, -24, -8), 0) + || test_slice_indices(a[i], IntArray(4, 20, 4), 1) + || test_slice_indices(a[i], IntArray(1, -12), 2); if (ret != 0) return ret; @@ -184,14 +196,14 @@ static int test_slice_2() for (int i = 0; i < sizeof(a) / sizeof(a[0]); i++) { int ret = 0 - || test_slice(a[i], IntArrayMat(-233, -233, -233), 0) - || test_slice(a[i], IntArrayMat(-233, -233, -233), -1) - || test_slice(a[i], IntArrayMat(3, 12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(32, 8, -233), -2) - || test_slice(a[i], IntArrayMat(2, 12, 16, -233), -1) - || test_slice_indices(a[i], IntArrayMat(2, -24, -8), 0) - || test_slice_indices(a[i], IntArrayMat(1, -12), 1); + || test_slice(a[i], IntArray(-233, -233, -233), 0) + || test_slice(a[i], IntArray(-233, -233, -233), -1) + || test_slice(a[i], IntArray(3, 12, 16, -233), 0) + || test_slice(a[i], IntArray(12, 16, -233), 0) + || test_slice(a[i], IntArray(32, 8, -233), -2) + || test_slice(a[i], IntArray(2, 12, 16, -233), -1) + || test_slice_indices(a[i], IntArray(2, -24, -8), 0) + || test_slice_indices(a[i], IntArray(1, -12), 1); if (ret != 0) return ret; @@ -211,11 +223,11 @@ static int test_slice_3() for (int i = 0; i < sizeof(a) / sizeof(a[0]); i++) { int ret = 0 - || test_slice(a[i], IntArrayMat(-233, -233, -233), 0) - || test_slice(a[i], IntArrayMat(3, 12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(12, 16, -233), 0) - || test_slice(a[i], IntArrayMat(32, 8, -233), -1) - || test_slice_indices(a[i], IntArrayMat(2, -24, -8), 0); + || test_slice(a[i], IntArray(-233, -233, -233), 0) + || test_slice(a[i], IntArray(3, 12, 16, -233), 0) + || test_slice(a[i], IntArray(12, 16, -233), 0) + || test_slice(a[i], IntArray(32, 8, -233), -1) + || test_slice_indices(a[i], IntArray(2, -24, -8), 0); if (ret != 0) return ret; diff --git a/tests/test_slice_oom.cpp b/tests/test_slice_oom.cpp index 62c717ba045..cabf56a2384 100644 --- a/tests/test_slice_oom.cpp +++ b/tests/test_slice_oom.cpp @@ -14,58 +14,61 @@ #include "testutil.h" -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_slice_oom(const ncnn::Mat& a, const ncnn::Mat& slices, int axis) +static int test_slice_oom(const ncnn::Mat& a, const std::vector& slices_array, int axis) { + ncnn::Mat slices(slices_array.size()); + { + int* p = slices; + for (size_t i = 0; i < slices_array.size(); i++) + { + p[i] = slices_array[i]; + } + } + ncnn::ParamDict pd; pd.set(0, slices); pd.set(1, axis); @@ -80,15 +83,24 @@ static int test_slice_oom(const ncnn::Mat& a, const ncnn::Mat& slices, int axis) { fprintf(stderr, "test_slice_oom failed a.dims=%d a=(%d %d %d %d)", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " slices="); - print_int_array(slices); + print_int_array(slices_array); fprintf(stderr, " axis=%d\n", axis); } return ret; } -static int test_slice_oom_indices(const ncnn::Mat& a, const ncnn::Mat& indices, int axis) +static int test_slice_oom_indices(const ncnn::Mat& a, const std::vector& indices_array, int axis) { + ncnn::Mat indices(indices_array.size()); + { + int* p = indices; + for (size_t i = 0; i < indices_array.size(); i++) + { + p[i] = indices_array[i]; + } + } + ncnn::ParamDict pd; pd.set(1, axis); pd.set(2, indices); @@ -103,7 +115,7 @@ static int test_slice_oom_indices(const ncnn::Mat& a, const ncnn::Mat& indices, { fprintf(stderr, "test_slice_oom_indices failed a.dims=%d a=(%d %d %d %d)", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " indices="); - print_int_array(indices); + print_int_array(indices_array); fprintf(stderr, " axis=%d\n", axis); } @@ -115,11 +127,11 @@ static int test_slice_0() ncnn::Mat a = RandomMat(48, 48, 48, 48); return 0 - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 0) - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 1) - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 2) - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 3) - || test_slice_oom_indices(a, IntArrayMat(2, -24, -8), 0); + || test_slice_oom(a, IntArray(3, 12, 16, -233), 0) + || test_slice_oom(a, IntArray(3, 12, 16, -233), 1) + || test_slice_oom(a, IntArray(3, 12, 16, -233), 2) + || test_slice_oom(a, IntArray(3, 12, 16, -233), 3) + || test_slice_oom_indices(a, IntArray(2, -24, -8), 0); } static int test_slice_1() @@ -127,10 +139,10 @@ static int test_slice_1() ncnn::Mat a = RandomMat(48, 48, 48); return 0 - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 0) - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 1) - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 2) - || test_slice_oom_indices(a, IntArrayMat(2, -24, -8), 0); + || test_slice_oom(a, IntArray(3, 12, 16, -233), 0) + || test_slice_oom(a, IntArray(3, 12, 16, -233), 1) + || test_slice_oom(a, IntArray(3, 12, 16, -233), 2) + || test_slice_oom_indices(a, IntArray(2, -24, -8), 0); } static int test_slice_2() @@ -138,9 +150,9 @@ static int test_slice_2() ncnn::Mat a = RandomMat(48, 48); return 0 - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 0) - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 1) - || test_slice_oom_indices(a, IntArrayMat(2, -24, -8), 0); + || test_slice_oom(a, IntArray(3, 12, 16, -233), 0) + || test_slice_oom(a, IntArray(3, 12, 16, -233), 1) + || test_slice_oom_indices(a, IntArray(2, -24, -8), 0); } static int test_slice_3() @@ -148,8 +160,8 @@ static int test_slice_3() ncnn::Mat a = RandomMat(48); return 0 - || test_slice_oom(a, IntArrayMat(3, 12, 16, -233), 0) - || test_slice_oom_indices(a, IntArrayMat(2, -24, -8), 0); + || test_slice_oom(a, IntArray(3, 12, 16, -233), 0) + || test_slice_oom_indices(a, IntArray(2, -24, -8), 0); } int main() diff --git a/tests/test_squeeze.cpp b/tests/test_squeeze.cpp index 02f772c8581..30df274ab68 100644 --- a/tests/test_squeeze.cpp +++ b/tests/test_squeeze.cpp @@ -33,58 +33,61 @@ static int test_squeeze(const ncnn::Mat& a, int squeeze_w, int squeeze_h, int sq return ret; } -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_squeeze_axes(const ncnn::Mat& a, const ncnn::Mat& axes) +static int test_squeeze_axes(const ncnn::Mat& a, const std::vector& axes_array) { + ncnn::Mat axes(axes_array.size()); + { + int* p = axes; + for (size_t i = 0; i < axes_array.size(); i++) + { + p[i] = axes_array[i]; + } + } + ncnn::ParamDict pd; pd.set(3, axes); @@ -95,7 +98,7 @@ static int test_squeeze_axes(const ncnn::Mat& a, const ncnn::Mat& axes) { fprintf(stderr, "test_squeeze_axes failed a.dims=%d a=(%d %d %d %d)\n", a.dims, a.w, a.h, a.d, a.c); fprintf(stderr, " axes="); - print_int_array(axes); + print_int_array(axes_array); fprintf(stderr, "\n"); } @@ -122,21 +125,21 @@ static int test_squeeze_all_params(const ncnn::Mat& a) || test_squeeze(a, 1, 1, 1, 0) || test_squeeze(a, 1, 1, 1, 1) - || test_squeeze_axes(a, IntArrayMat(0)) - || test_squeeze_axes(a, IntArrayMat(1)) - || test_squeeze_axes(a, IntArrayMat(2)) - || test_squeeze_axes(a, IntArrayMat(3)) - || test_squeeze_axes(a, IntArrayMat(0, 1)) - || test_squeeze_axes(a, IntArrayMat(0, 2)) - || test_squeeze_axes(a, IntArrayMat(0, 3)) - || test_squeeze_axes(a, IntArrayMat(1, 2)) - || test_squeeze_axes(a, IntArrayMat(1, 3)) - || test_squeeze_axes(a, IntArrayMat(2, 3)) - || test_squeeze_axes(a, IntArrayMat(0, 1, 2)) - || test_squeeze_axes(a, IntArrayMat(0, 1, 3)) - || test_squeeze_axes(a, IntArrayMat(0, 2, 3)) - || test_squeeze_axes(a, IntArrayMat(1, 2, 3)) - || test_squeeze_axes(a, IntArrayMat(0, 1, 2, 3)); + || test_squeeze_axes(a, IntArray(0)) + || test_squeeze_axes(a, IntArray(1)) + || test_squeeze_axes(a, IntArray(2)) + || test_squeeze_axes(a, IntArray(3)) + || test_squeeze_axes(a, IntArray(0, 1)) + || test_squeeze_axes(a, IntArray(0, 2)) + || test_squeeze_axes(a, IntArray(0, 3)) + || test_squeeze_axes(a, IntArray(1, 2)) + || test_squeeze_axes(a, IntArray(1, 3)) + || test_squeeze_axes(a, IntArray(2, 3)) + || test_squeeze_axes(a, IntArray(0, 1, 2)) + || test_squeeze_axes(a, IntArray(0, 1, 3)) + || test_squeeze_axes(a, IntArray(0, 2, 3)) + || test_squeeze_axes(a, IntArray(1, 2, 3)) + || test_squeeze_axes(a, IntArray(0, 1, 2, 3)); } static int test_squeeze_0() diff --git a/tests/test_tile.cpp b/tests/test_tile.cpp index ffc238eb10c..f663f35c875 100644 --- a/tests/test_tile.cpp +++ b/tests/test_tile.cpp @@ -31,58 +31,61 @@ static int test_tile(const ncnn::Mat& a, int axis, int tiles) return ret; } -static ncnn::Mat IntArrayMat(int a0) +static std::vector IntArray(int a0) { - ncnn::Mat m(1); - int* p = m; - p[0] = a0; + std::vector m(1); + m[0] = a0; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1) +static std::vector IntArray(int a0, int a1) { - ncnn::Mat m(2); - int* p = m; - p[0] = a0; - p[1] = a1; + std::vector m(2); + m[0] = a0; + m[1] = a1; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2) +static std::vector IntArray(int a0, int a1, int a2) { - ncnn::Mat m(3); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; + std::vector m(3); + m[0] = a0; + m[1] = a1; + m[2] = a2; return m; } -static ncnn::Mat IntArrayMat(int a0, int a1, int a2, int a3) +static std::vector IntArray(int a0, int a1, int a2, int a3) { - ncnn::Mat m(4); - int* p = m; - p[0] = a0; - p[1] = a1; - p[2] = a2; - p[3] = a3; + std::vector m(4); + m[0] = a0; + m[1] = a1; + m[2] = a2; + m[3] = a3; return m; } -static void print_int_array(const ncnn::Mat& a) +static void print_int_array(const std::vector& a) { - const int* pa = a; - fprintf(stderr, "["); - for (int i = 0; i < a.w; i++) + for (size_t i = 0; i < a.size(); i++) { - fprintf(stderr, " %d", pa[i]); + fprintf(stderr, " %d", a[i]); } fprintf(stderr, " ]"); } -static int test_tile(const ncnn::Mat& a, const ncnn::Mat& repeats) +static int test_tile(const ncnn::Mat& a, const std::vector& repeats_array) { + ncnn::Mat repeats(repeats_array.size()); + { + int* p = repeats; + for (size_t i = 0; i < repeats_array.size(); i++) + { + p[i] = repeats_array[i]; + } + } + ncnn::ParamDict pd; pd.set(2, repeats); @@ -92,7 +95,7 @@ static int test_tile(const ncnn::Mat& a, const ncnn::Mat& repeats) if (ret != 0) { fprintf(stderr, "test_tile failed a.dims=%d a=(%d %d %d %d) repeats=", a.dims, a.w, a.h, a.d, a.c); - print_int_array(repeats); + print_int_array(repeats_array); fprintf(stderr, "\n"); } @@ -119,18 +122,18 @@ static int test_tile_0() || test_tile(c, 2, 5) || test_tile(c, 3, 2) - || test_tile(a, IntArrayMat(3)) - || test_tile(a, IntArrayMat(2, 4)) - || test_tile(a, IntArrayMat(2, 2, 5)) - || test_tile(a, IntArrayMat(3, 1, 3, 2)) - || test_tile(b, IntArrayMat(3, 1)) - || test_tile(b, IntArrayMat(4, 1, 4)) - || test_tile(b, IntArrayMat(2, 2, 2, 1)) - || test_tile(b, IntArrayMat(3, 2, 1)) - || test_tile(c, IntArrayMat(3)) - || test_tile(c, IntArrayMat(1, 1, 4)) - || test_tile(c, IntArrayMat(2, 2, 5)) - || test_tile(c, IntArrayMat(3, 2, 1, 9)); + || test_tile(a, IntArray(3)) + || test_tile(a, IntArray(2, 4)) + || test_tile(a, IntArray(2, 2, 5)) + || test_tile(a, IntArray(3, 1, 3, 2)) + || test_tile(b, IntArray(3, 1)) + || test_tile(b, IntArray(4, 1, 4)) + || test_tile(b, IntArray(2, 2, 2, 1)) + || test_tile(b, IntArray(3, 2, 1)) + || test_tile(c, IntArray(3)) + || test_tile(c, IntArray(1, 1, 4)) + || test_tile(c, IntArray(2, 2, 5)) + || test_tile(c, IntArray(3, 2, 1, 9)); } static int test_tile_1() @@ -150,18 +153,18 @@ static int test_tile_1() || test_tile(c, 1, 2) || test_tile(c, 2, 2) - || test_tile(a, IntArrayMat(5)) - || test_tile(a, IntArrayMat(1, 4)) - || test_tile(a, IntArrayMat(2, 1, 4)) - || test_tile(a, IntArrayMat(1, 2, 1, 4)) - || test_tile(b, IntArrayMat(3)) - || test_tile(b, IntArrayMat(1, 3, 3)) - || test_tile(b, IntArrayMat(2, 3)) - || test_tile(b, IntArrayMat(2, 3, 3, 3)) - || test_tile(c, IntArrayMat(1)) - || test_tile(c, IntArrayMat(2, 1)) - || test_tile(c, IntArrayMat(2, 2, 2)) - || test_tile(c, IntArrayMat(2, 1, 2, 1)); + || test_tile(a, IntArray(5)) + || test_tile(a, IntArray(1, 4)) + || test_tile(a, IntArray(2, 1, 4)) + || test_tile(a, IntArray(1, 2, 1, 4)) + || test_tile(b, IntArray(3)) + || test_tile(b, IntArray(1, 3, 3)) + || test_tile(b, IntArray(2, 3)) + || test_tile(b, IntArray(2, 3, 3, 3)) + || test_tile(c, IntArray(1)) + || test_tile(c, IntArray(2, 1)) + || test_tile(c, IntArray(2, 2, 2)) + || test_tile(c, IntArray(2, 1, 2, 1)); } static int test_tile_2() @@ -178,18 +181,18 @@ static int test_tile_2() || test_tile(c, 0, 5) || test_tile(c, 1, 6) - || test_tile(a, IntArrayMat(2)) - || test_tile(a, IntArrayMat(1, 1)) - || test_tile(a, IntArrayMat(4, 1, 1)) - || test_tile(a, IntArrayMat(2, 4, 4, 1)) - || test_tile(b, IntArrayMat(3)) - || test_tile(b, IntArrayMat(2, 4)) - || test_tile(b, IntArrayMat(2, 4, 3, 1)) - || test_tile(b, IntArrayMat(1, 2, 1, 4)) - || test_tile(c, IntArrayMat(5)) - || test_tile(c, IntArrayMat(6, 1)) - || test_tile(c, IntArrayMat(6, 1, 6)) - || test_tile(c, IntArrayMat(3, 2, 1, 1)); + || test_tile(a, IntArray(2)) + || test_tile(a, IntArray(1, 1)) + || test_tile(a, IntArray(4, 1, 1)) + || test_tile(a, IntArray(2, 4, 4, 1)) + || test_tile(b, IntArray(3)) + || test_tile(b, IntArray(2, 4)) + || test_tile(b, IntArray(2, 4, 3, 1)) + || test_tile(b, IntArray(1, 2, 1, 4)) + || test_tile(c, IntArray(5)) + || test_tile(c, IntArray(6, 1)) + || test_tile(c, IntArray(6, 1, 6)) + || test_tile(c, IntArray(3, 2, 1, 1)); } static int test_tile_3() @@ -204,20 +207,20 @@ static int test_tile_3() || test_tile(b, 0, 3) || test_tile(c, 0, 4) - || test_tile(a, IntArrayMat(10)) - || test_tile(a, IntArrayMat(10, 1)) - || test_tile(a, IntArrayMat(5, 2, 1)) - || test_tile(a, IntArrayMat(2, 2, 2, 3)) - || test_tile(b, IntArrayMat(2)) - || test_tile(b, IntArrayMat(2, 2)) - || test_tile(b, IntArrayMat(2, 2, 1)) - || test_tile(b, IntArrayMat(4, 1, 2, 2)) - || test_tile(c, IntArrayMat(3)) - || test_tile(c, IntArrayMat(4, 3)) - || test_tile(c, IntArrayMat(1)) - || test_tile(c, IntArrayMat(1, 1)) - || test_tile(c, IntArrayMat(1, 1, 1)) - || test_tile(c, IntArrayMat(1, 3, 2, 2)); + || test_tile(a, IntArray(10)) + || test_tile(a, IntArray(10, 1)) + || test_tile(a, IntArray(5, 2, 1)) + || test_tile(a, IntArray(2, 2, 2, 3)) + || test_tile(b, IntArray(2)) + || test_tile(b, IntArray(2, 2)) + || test_tile(b, IntArray(2, 2, 1)) + || test_tile(b, IntArray(4, 1, 2, 2)) + || test_tile(c, IntArray(3)) + || test_tile(c, IntArray(4, 3)) + || test_tile(c, IntArray(1)) + || test_tile(c, IntArray(1, 1)) + || test_tile(c, IntArray(1, 1, 1)) + || test_tile(c, IntArray(1, 3, 2, 2)); } int main()