nihui
/
ncnn
Not watched
1
0
Fork 0
Code
Releases 46 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
ncnn/tests/test_matmul.cpp

244 lines
8.3 kB
Raw Permalink Blame History 

  1. // Copyright 2022 Tencent

  2. // SPDX-License-Identifier: BSD-3-Clause

  3. 

  4. #include "testutil.h"

  5. 

  6. static int test_matmul(const ncnn::Mat& a, const ncnn::Mat& b)

  7. {

  8.     ncnn::ParamDict pd;

  9.     pd.set(0, 0); // transB

  10. 

  11.     std::vector<ncnn::Mat> weights(0);

  12. 

  13.     std::vector<ncnn::Mat> as(2);

  14.     as[0] = a;

  15.     as[1] = b;

  16. 

  17.     int ret = test_layer("MatMul", pd, weights, as);

  18.     if (ret != 0)

  19.     {

  20.         fprintf(stderr, "test_matmul failed a.dims=%d a=(%d %d %d %d) b.dims=%d b=(%d %d %d %d)\n", a.dims, a.w, a.h, a.d, a.c, b.dims, b.w, b.h, b.d, b.c);

  21.     }

  22. 

  23.     return ret;

  24. }

  25. 

  26. static int test_matmul_transb(const ncnn::Mat& a, const ncnn::Mat& b)

  27. {

  28.     ncnn::ParamDict pd;

  29.     pd.set(0, 1); // transB

  30. 

  31.     std::vector<ncnn::Mat> weights(0);

  32. 

  33.     std::vector<ncnn::Mat> as(2);

  34.     as[0] = a;

  35.     as[1] = b;

  36. 

  37.     int ret = test_layer("MatMul", pd, weights, as);

  38.     if (ret != 0)

  39.     {

  40.         fprintf(stderr, "test_matmul_transb failed a.dims=%d a=(%d %d %d %d) b.dims=%d b=(%d %d %d %d)\n", a.dims, a.w, a.h, a.d, a.c, b.dims, b.w, b.h, b.d, b.c);

  41.     }

  42. 

  43.     return ret;

  44. }

  45. 

  46. static int test_matmul_0()

  47. {

  48.     return 0

  49.            || test_matmul(RandomMat(124), RandomMat(124))

  50.            || test_matmul(RandomMat(127), RandomMat(127))

  51.            || test_matmul(RandomMat(128), RandomMat(128));

  52. }

  53. 

  54. static int test_matmul_1()

  55. {

  56.     return 0

  57.            || test_matmul(RandomMat(5), RandomMat(6, 5))

  58.            || test_matmul(RandomMat(16), RandomMat(12, 16))

  59.            || test_matmul(RandomMat(11), RandomMat(16, 11))

  60. 

  61.            || test_matmul_transb(RandomMat(5), RandomMat(5, 6))

  62.            || test_matmul_transb(RandomMat(16), RandomMat(16, 12))

  63.            || test_matmul_transb(RandomMat(11), RandomMat(11, 16));

  64. }

  65. 

  66. static int test_matmul_2()

  67. {

  68.     return 0

  69.            || test_matmul(RandomMat(13), RandomMat(7, 13, 12))

  70.            || test_matmul(RandomMat(24), RandomMat(6, 24, 16))

  71.            || test_matmul(RandomMat(20), RandomMat(8, 20, 19))

  72. 

  73.            || test_matmul_transb(RandomMat(13), RandomMat(13, 7, 12))

  74.            || test_matmul_transb(RandomMat(24), RandomMat(24, 6, 16))

  75.            || test_matmul_transb(RandomMat(20), RandomMat(20, 8, 19));

  76. }

  77. 

  78. static int test_matmul_3()

  79. {

  80.     return 0

  81.            || test_matmul(RandomMat(13), RandomMat(7, 13, 5, 12))

  82.            || test_matmul(RandomMat(24), RandomMat(6, 24, 4, 16))

  83.            || test_matmul(RandomMat(20), RandomMat(8, 20, 3, 19))

  84. 

  85.            || test_matmul_transb(RandomMat(13), RandomMat(13, 7, 5, 12))

  86.            || test_matmul_transb(RandomMat(24), RandomMat(24, 6, 4, 16))

  87.            || test_matmul_transb(RandomMat(20), RandomMat(20, 8, 3, 19));

  88. }

  89. 

  90. static int test_matmul_4()

  91. {

  92.     return 0

  93.            || test_matmul(RandomMat(5, 6), RandomMat(5))

  94.            || test_matmul(RandomMat(16, 12), RandomMat(16))

  95.            || test_matmul(RandomMat(11, 16), RandomMat(11));

  96. }

  97. 

  98. static int test_matmul_5()

  99. {

  100.     return 0

  101.            || test_matmul(RandomMat(32, 3, 10), RandomMat(32))

  102.            || test_matmul(RandomMat(31, 4, 16), RandomMat(31))

  103.            || test_matmul(RandomMat(28, 5, 28), RandomMat(28));

  104. }

  105. 

  106. static int test_matmul_6()

  107. {

  108.     return 0

  109.            || test_matmul(RandomMat(32, 3, 4, 10), RandomMat(32))

  110.            || test_matmul(RandomMat(31, 4, 5, 16), RandomMat(31))

  111.            || test_matmul(RandomMat(18, 5, 6, 28), RandomMat(18));

  112. }

  113. 

  114. static int test_matmul_7()

  115. {

  116.     return 0

  117.            || test_matmul(RandomMat(14, 10), RandomMat(5, 14))

  118.            || test_matmul(RandomMat(16, 16), RandomMat(10, 16))

  119.            || test_matmul(RandomMat(14, 28), RandomMat(9, 14))

  120. 

  121.            || test_matmul_transb(RandomMat(14, 10), RandomMat(14, 5))

  122.            || test_matmul_transb(RandomMat(16, 16), RandomMat(16, 10))

  123.            || test_matmul_transb(RandomMat(14, 28), RandomMat(14, 9));

  124. }

  125. 

  126. static int test_matmul_8()

  127. {

  128.     return 0

  129.            || test_matmul(RandomMat(5, 4), RandomMat(4, 5, 12))

  130.            || test_matmul(RandomMat(5, 14), RandomMat(5, 5, 16))

  131.            || test_matmul(RandomMat(5, 24), RandomMat(6, 5, 19))

  132. 

  133.            || test_matmul_transb(RandomMat(5, 4), RandomMat(5, 4, 12))

  134.            || test_matmul_transb(RandomMat(5, 14), RandomMat(5, 5, 16))

  135.            || test_matmul_transb(RandomMat(5, 24), RandomMat(5, 6, 19));

  136. }

  137. 

  138. static int test_matmul_9()

  139. {

  140.     return 0

  141.            || test_matmul(RandomMat(5, 4), RandomMat(4, 5, 2, 12))

  142.            || test_matmul(RandomMat(5, 14), RandomMat(5, 5, 3, 16))

  143.            || test_matmul(RandomMat(5, 24), RandomMat(6, 5, 4, 19))

  144. 

  145.            || test_matmul_transb(RandomMat(5, 4), RandomMat(5, 4, 2, 12))

  146.            || test_matmul_transb(RandomMat(5, 14), RandomMat(5, 5, 3, 16))

  147.            || test_matmul_transb(RandomMat(5, 24), RandomMat(5, 6, 4, 19));

  148. }

  149. 

  150. static int test_matmul_10()

  151. {

  152.     return 0

  153.            || test_matmul(RandomMat(14, 23, 10), RandomMat(5, 14))

  154.            || test_matmul(RandomMat(16, 22, 16), RandomMat(10, 16))

  155.            || test_matmul(RandomMat(14, 20, 28), RandomMat(9, 14))

  156. 

  157.            || test_matmul_transb(RandomMat(14, 23, 10), RandomMat(14, 5))

  158.            || test_matmul_transb(RandomMat(16, 22, 16), RandomMat(16, 10))

  159.            || test_matmul_transb(RandomMat(14, 20, 28), RandomMat(14, 9));

  160. }

  161. 

  162. static int test_matmul_11()

  163. {

  164.     return 0

  165.            || test_matmul(RandomMat(14, 13, 2, 10), RandomMat(5, 14))

  166.            || test_matmul(RandomMat(16, 12, 3, 16), RandomMat(10, 16))

  167.            || test_matmul(RandomMat(14, 10, 4, 28), RandomMat(9, 14))

  168. 

  169.            || test_matmul_transb(RandomMat(14, 13, 2, 10), RandomMat(14, 5))

  170.            || test_matmul_transb(RandomMat(16, 12, 3, 16), RandomMat(16, 10))

  171.            || test_matmul_transb(RandomMat(14, 10, 4, 28), RandomMat(14, 9));

  172. }

  173. 

  174. static int test_matmul_12()

  175. {

  176.     return 0

  177.            || test_matmul(RandomMat(14, 23, 10), RandomMat(5, 14, 10))

  178.            || test_matmul(RandomMat(16, 22, 16), RandomMat(10, 16, 16))

  179.            || test_matmul(RandomMat(14, 20, 28), RandomMat(9, 14, 28))

  180. 

  181.            || test_matmul_transb(RandomMat(14, 23, 10), RandomMat(14, 5, 10))

  182.            || test_matmul_transb(RandomMat(16, 22, 16), RandomMat(16, 10, 16))

  183.            || test_matmul_transb(RandomMat(14, 20, 28), RandomMat(14, 9, 28));

  184. }

  185. 

  186. static int test_matmul_13()

  187. {

  188.     return 0

  189.            || test_matmul(RandomMat(14, 23, 10), RandomMat(5, 14, 1, 16))

  190.            || test_matmul(RandomMat(16, 22, 9), RandomMat(10, 16, 1, 17))

  191.            || test_matmul(RandomMat(14, 20, 8), RandomMat(9, 14, 1, 18))

  192. 

  193.            || test_matmul_transb(RandomMat(14, 23, 10), RandomMat(14, 5, 1, 16))

  194.            || test_matmul_transb(RandomMat(16, 22, 9), RandomMat(16, 10, 1, 17))

  195.            || test_matmul_transb(RandomMat(14, 20, 8), RandomMat(14, 9, 1, 18));

  196. }

  197. 

  198. static int test_matmul_14()

  199. {

  200.     return 0

  201.            || test_matmul(RandomMat(14, 23, 10, 1), RandomMat(5, 14, 1, 16))

  202.            || test_matmul(RandomMat(16, 22, 9, 1), RandomMat(10, 16, 1, 17))

  203.            || test_matmul(RandomMat(14, 20, 8, 1), RandomMat(9, 14, 1, 18))

  204. 

  205.            || test_matmul_transb(RandomMat(14, 23, 10, 1), RandomMat(14, 5, 1, 16))

  206.            || test_matmul_transb(RandomMat(16, 22, 9, 1), RandomMat(16, 10, 1, 17))

  207.            || test_matmul_transb(RandomMat(14, 20, 8, 1), RandomMat(14, 9, 1, 18));

  208. }

  209. 

  210. static int test_matmul_15()

  211. {

  212.     return 0

  213.            || test_matmul(RandomMat(14, 23, 10, 16), RandomMat(5, 14, 10, 16))

  214.            || test_matmul(RandomMat(16, 22, 9, 17), RandomMat(10, 16, 9, 17))

  215.            || test_matmul(RandomMat(14, 20, 8, 18), RandomMat(9, 14, 8, 18))

  216. 

  217.            || test_matmul_transb(RandomMat(14, 23, 10, 16), RandomMat(14, 5, 10, 16))

  218.            || test_matmul_transb(RandomMat(16, 22, 9, 17), RandomMat(16, 10, 9, 17))

  219.            || test_matmul_transb(RandomMat(14, 20, 8, 18), RandomMat(14, 9, 8, 18));

  220. }

  221. 

  222. int main()

  223. {

  224.     SRAND(7767517);

  225. 

  226.     return 0

  227.            || test_matmul_0()

  228.            || test_matmul_1()

  229.            || test_matmul_2()

  230.            || test_matmul_3()

  231.            || test_matmul_4()

  232.            || test_matmul_5()

  233.            || test_matmul_6()

  234.            || test_matmul_7()

  235.            || test_matmul_8()

  236.            || test_matmul_9()

  237.            || test_matmul_10()

  238.            || test_matmul_11()

  239.            || test_matmul_12()

  240.            || test_matmul_13()

  241.            || test_matmul_14()

  242.            || test_matmul_15();

  243. }