Home Explore Help
Register Sign In
vimacs
/
riscv-tests
Watch 1
Star 0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Branch: vimacs
Branches Tags
riscv-tests
/
mt
/
av_matmul.c

av_matmul.c 8.0 KB
Permalink History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260 
  1. #include "stdlib.h"
    2. 

  2. 

  3. #include "util.h"
    4. 

  4. 

  5. #include "dataset.h"
    6. 

  6. void __attribute__((noinline)) matmul(const int coreid, const int ncores, const int lda,  const data_t A[], const data_t B[], data_t C[] )
    7. 

  7. {
    8. 

  8.    
    9. 

  9. 	//-----------------------------------------------------------------version 2.16, optimize v2.15 get rid of tempb. MSI 83K.w/ test one 81K.
    10. 

  10. 

  11. 	
    12. 

  12. 	static __thread data_t TempA[8];
    13. 

  13. 	static __thread data_t TempB[8];
    14. 

  14. 	static __thread data_t TempC[8];
    15. 

  15. 	static __thread int j,m,n;
    16. 

  16. 	
    17. 

  17. 			if(coreid == 1 || ncores == 1 )
    18. 

  18. 	{
    19. 

  19. 	for ( j = 16; j < 32; j++ )
    20. 

  20.      {		
    21. 

  21. 		  
    22. 

  22.       for ( m = 0; m < 4; m++ )  
    23. 

  23.       {
    24. 

  24. 		  
    25. 

  25. 		 TempA[0] = A[j*lda+0+8*m];
    26. 

  26. 		 TempA[1] = A[j*lda+1+8*m];
    27. 

  27. 		 TempA[2] = A[j*lda+2+8*m];
    28. 

  28. 		 TempA[3] = A[j*lda+3+8*m];
    29. 

  29. 		 TempA[4] = A[j*lda+4+8*m];
    30. 

  30. 		 TempA[5] = A[j*lda+5+8*m];
    31. 

  31. 		 TempA[6] = A[j*lda+6+8*m];
    32. 

  32. 		 TempA[7] = A[j*lda+7+8*m];
    33. 

  33. 		 
    34. 

  34. 

  35. 		 
    36. 

  36. 		for( n = 0; n < 4; n++)
    37. 

  37. 		{
    38. 

  38. 	     
    39. 

  39. 		 
    40. 

  40. 

  41. 		 
    42. 

  42. 		 
    43. 

  43. 		 TempC[0] = TempA[0] * B[(0+8*m)*lda+0+8*n]; 
    44. 

  44. 		 TempC[1] = TempA[0] * B[(0+8*m)*lda+1+8*n]; 
    45. 

  45. 		 TempC[2] = TempA[0] * B[(0+8*m)*lda+2+8*n]; 
    46. 

  46. 		 TempC[3] = TempA[0] * B[(0+8*m)*lda+3+8*n]; 
    47. 

  47. 		 TempC[4] = TempA[0] * B[(0+8*m)*lda+4+8*n]; 
    48. 

  48. 		 TempC[5] = TempA[0] * B[(0+8*m)*lda+5+8*n]; 
    49. 

  49. 		 TempC[6] = TempA[0] * B[(0+8*m)*lda+6+8*n]; 
    50. 

  50. 		 TempC[7] = TempA[0] * B[(0+8*m)*lda+7+8*n]; 
    51. 

  51. 		 
    52. 

  52. 		 
    53. 

  53. 		 TempC[0] += TempA[1] * B[(1+8*m)*lda+0+8*n]; 
    54. 

  54. 		 TempC[1] += TempA[1] * B[(1+8*m)*lda+1+8*n]; 
    55. 

  55. 		 TempC[2] += TempA[1] * B[(1+8*m)*lda+2+8*n]; 
    56. 

  56. 		 TempC[3] += TempA[1] * B[(1+8*m)*lda+3+8*n]; 
    57. 

  57. 		 TempC[4] += TempA[1] * B[(1+8*m)*lda+4+8*n]; 
    58. 

  58. 		 TempC[5] += TempA[1] * B[(1+8*m)*lda+5+8*n]; 
    59. 

  59. 		 TempC[6] += TempA[1] * B[(1+8*m)*lda+6+8*n]; 
    60. 

  60. 		 TempC[7] += TempA[1] * B[(1+8*m)*lda+7+8*n]; 
    61. 

  61. 		 
    62. 

  62. 

  63. 

  64. 		 TempC[0] += TempA[2] * B[(2+8*m)*lda+0+8*n]; 
    65. 

  65. 		 TempC[1] += TempA[2] * B[(2+8*m)*lda+1+8*n]; 
    66. 

  66. 		 TempC[2] += TempA[2] * B[(2+8*m)*lda+2+8*n]; 
    67. 

  67. 		 TempC[3] += TempA[2] * B[(2+8*m)*lda+3+8*n]; 
    68. 

  68. 		 TempC[4] += TempA[2] * B[(2+8*m)*lda+4+8*n]; 
    69. 

  69. 		 TempC[5] += TempA[2] * B[(2+8*m)*lda+5+8*n]; 
    70. 

  70. 		 TempC[6] += TempA[2] * B[(2+8*m)*lda+6+8*n]; 
    71. 

  71. 		 TempC[7] += TempA[2] * B[(2+8*m)*lda+7+8*n]; 
    72. 

  72. 		 
    73. 

  73. 

  74. 

  75. 		 TempC[0] += TempA[3] * B[(3+8*m)*lda+0+8*n]; 
    76. 

  76. 		 TempC[1] += TempA[3] * B[(3+8*m)*lda+1+8*n]; 
    77. 

  77. 		 TempC[2] += TempA[3] * B[(3+8*m)*lda+2+8*n]; 
    78. 

  78. 		 TempC[3] += TempA[3] * B[(3+8*m)*lda+3+8*n]; 
    79. 

  79. 		 TempC[4] += TempA[3] * B[(3+8*m)*lda+4+8*n]; 
    80. 

  80. 		 TempC[5] += TempA[3] * B[(3+8*m)*lda+5+8*n]; 
    81. 

  81. 		 TempC[6] += TempA[3] * B[(3+8*m)*lda+6+8*n]; 
    82. 

  82. 		 TempC[7] += TempA[3] * B[(3+8*m)*lda+7+8*n]; 
    83. 

  83. 

  84. 		 TempC[0] += TempA[4] * B[(4+8*m)*lda+0+8*n]; 
    85. 

  85. 		 TempC[1] += TempA[4] * B[(4+8*m)*lda+1+8*n]; 
    86. 

  86. 		 TempC[2] += TempA[4] * B[(4+8*m)*lda+2+8*n]; 
    87. 

  87. 		 TempC[3] += TempA[4] * B[(4+8*m)*lda+3+8*n]; 
    88. 

  88. 		 TempC[4] += TempA[4] * B[(4+8*m)*lda+4+8*n]; 
    89. 

  89. 		 TempC[5] += TempA[4] * B[(4+8*m)*lda+5+8*n]; 
    90. 

  90. 		 TempC[6] += TempA[4] * B[(4+8*m)*lda+6+8*n]; 
    91. 

  91. 		 TempC[7] += TempA[4] * B[(4+8*m)*lda+7+8*n]; 
    92. 

  92. 

  93. 

  94. 		 TempC[0] += TempA[5] * B[(5+8*m)*lda+0+8*n]; 
    95. 

  95. 		 TempC[1] += TempA[5] * B[(5+8*m)*lda+1+8*n]; 
    96. 

  96. 		 TempC[2] += TempA[5] * B[(5+8*m)*lda+2+8*n]; 
    97. 

  97. 		 TempC[3] += TempA[5] * B[(5+8*m)*lda+3+8*n]; 
    98. 

  98. 		 TempC[4] += TempA[5] * B[(5+8*m)*lda+4+8*n]; 
    99. 

  99. 		 TempC[5] += TempA[5] * B[(5+8*m)*lda+5+8*n]; 
    100. 

  100. 		 TempC[6] += TempA[5] * B[(5+8*m)*lda+6+8*n]; 
    101. 

  101. 		 TempC[7] += TempA[5] * B[(5+8*m)*lda+7+8*n]; 
    102. 

  102. 		 
    103. 

  103. 

  104. 

  105. 		 TempC[0] += TempA[6] * B[(6+8*m)*lda+0+8*n]; 
    106. 

  106. 		 TempC[1] += TempA[6] * B[(6+8*m)*lda+1+8*n]; 
    107. 

  107. 		 TempC[2] += TempA[6] * B[(6+8*m)*lda+2+8*n]; 
    108. 

  108. 		 TempC[3] += TempA[6] * B[(6+8*m)*lda+3+8*n]; 
    109. 

  109. 		 TempC[4] += TempA[6] * B[(6+8*m)*lda+4+8*n]; 
    110. 

  110. 		 TempC[5] += TempA[6] * B[(6+8*m)*lda+5+8*n]; 
    111. 

  111. 		 TempC[6] += TempA[6] * B[(6+8*m)*lda+6+8*n]; 
    112. 

  112. 		 TempC[7] += TempA[6] * B[(6+8*m)*lda+7+8*n]; 
    113. 

  113. 

  114. 

  115. 		 TempC[0] += TempA[7] * B[(7+8*m)*lda+0+8*n]; 
    116. 

  116. 		 TempC[1] += TempA[7] * B[(7+8*m)*lda+1+8*n]; 
    117. 

  117. 		 TempC[2] += TempA[7] * B[(7+8*m)*lda+2+8*n]; 
    118. 

  118. 		 TempC[3] += TempA[7] * B[(7+8*m)*lda+3+8*n]; 
    119. 

  119. 		 TempC[4] += TempA[7] * B[(7+8*m)*lda+4+8*n]; 
    120. 

  120. 		 TempC[5] += TempA[7] * B[(7+8*m)*lda+5+8*n]; 
    121. 

  121. 		 TempC[6] += TempA[7] * B[(7+8*m)*lda+6+8*n]; 
    122. 

  122. 		 TempC[7] += TempA[7] * B[(7+8*m)*lda+7+8*n]; 
    123. 

  123. 		 
    124. 

  124. 

  125. 		
    126. 

  126. 		 C[0+8*n+j*lda] += TempC[0];
    127. 

  127. 		 C[1+8*n+j*lda] += TempC[1];
    128. 

  128. 		 C[2+8*n+j*lda] += TempC[2];
    129. 

  129. 		 C[3+8*n+j*lda] += TempC[3];
    130. 

  130. 		 C[4+8*n+j*lda] += TempC[4];
    131. 

  131. 		 C[5+8*n+j*lda] += TempC[5];
    132. 

  132. 		 C[6+8*n+j*lda] += TempC[6];
    133. 

  133. 		 C[7+8*n+j*lda] += TempC[7];
    134. 

  134. 		 }
    135. 

  135.       }
    136. 

  136. 	 }
    137. 

  137. 	}
    138. 

  138. 			if(coreid == 0)
    139. 

  139. 	{
    140. 

  140. 	for ( j = 0; j < 16; j++ )
    141. 

  141.      {		
    142. 

  142. 		  
    143. 

  143.       for ( m = 0; m < 4; m++ )  
    144. 

  144.       {
    145. 

  145. 		  
    146. 

  146. 		 TempA[0] = A[j*lda+0+8*m];
    147. 

  147. 		 TempA[1] = A[j*lda+1+8*m];
    148. 

  148. 		 TempA[2] = A[j*lda+2+8*m];
    149. 

  149. 		 TempA[3] = A[j*lda+3+8*m];
    150. 

  150. 		 TempA[4] = A[j*lda+4+8*m];
    151. 

  151. 		 TempA[5] = A[j*lda+5+8*m];
    152. 

  152. 		 TempA[6] = A[j*lda+6+8*m];
    153. 

  153. 		 TempA[7] = A[j*lda+7+8*m];
    154. 

  154. 		 
    155. 

  155. 

  156. 		 
    157. 

  157. 		for( n = 0; n < 4; n++)
    158. 

  158. 		{
    159. 

  159. 	     
    160. 

  160. 		 
    161. 

  161. 

  162. 		 
    163. 

  163. 		 
    164. 

  164. 		 TempC[0] = TempA[0] * B[(0+8*m)*lda+0+8*n]; 
    165. 

  165. 		 TempC[1] = TempA[0] * B[(0+8*m)*lda+1+8*n]; 
    166. 

  166. 		 TempC[2] = TempA[0] * B[(0+8*m)*lda+2+8*n]; 
    167. 

  167. 		 TempC[3] = TempA[0] * B[(0+8*m)*lda+3+8*n]; 
    168. 

  168. 		 TempC[4] = TempA[0] * B[(0+8*m)*lda+4+8*n]; 
    169. 

  169. 		 TempC[5] = TempA[0] * B[(0+8*m)*lda+5+8*n]; 
    170. 

  170. 		 TempC[6] = TempA[0] * B[(0+8*m)*lda+6+8*n]; 
    171. 

  171. 		 TempC[7] = TempA[0] * B[(0+8*m)*lda+7+8*n]; 
    172. 

  172. 		 
    173. 

  173. 		 
    174. 

  174. 		 TempC[0] += TempA[1] * B[(1+8*m)*lda+0+8*n]; 
    175. 

  175. 		 TempC[1] += TempA[1] * B[(1+8*m)*lda+1+8*n]; 
    176. 

  176. 		 TempC[2] += TempA[1] * B[(1+8*m)*lda+2+8*n]; 
    177. 

  177. 		 TempC[3] += TempA[1] * B[(1+8*m)*lda+3+8*n]; 
    178. 

  178. 		 TempC[4] += TempA[1] * B[(1+8*m)*lda+4+8*n]; 
    179. 

  179. 		 TempC[5] += TempA[1] * B[(1+8*m)*lda+5+8*n]; 
    180. 

  180. 		 TempC[6] += TempA[1] * B[(1+8*m)*lda+6+8*n]; 
    181. 

  181. 		 TempC[7] += TempA[1] * B[(1+8*m)*lda+7+8*n]; 
    182. 

  182. 		 
    183. 

  183. 

  184. 

  185. 		 TempC[0] += TempA[2] * B[(2+8*m)*lda+0+8*n]; 
    186. 

  186. 		 TempC[1] += TempA[2] * B[(2+8*m)*lda+1+8*n]; 
    187. 

  187. 		 TempC[2] += TempA[2] * B[(2+8*m)*lda+2+8*n]; 
    188. 

  188. 		 TempC[3] += TempA[2] * B[(2+8*m)*lda+3+8*n]; 
    189. 

  189. 		 TempC[4] += TempA[2] * B[(2+8*m)*lda+4+8*n]; 
    190. 

  190. 		 TempC[5] += TempA[2] * B[(2+8*m)*lda+5+8*n]; 
    191. 

  191. 		 TempC[6] += TempA[2] * B[(2+8*m)*lda+6+8*n]; 
    192. 

  192. 		 TempC[7] += TempA[2] * B[(2+8*m)*lda+7+8*n]; 
    193. 

  193. 		 
    194. 

  194. 

  195. 

  196. 		 TempC[0] += TempA[3] * B[(3+8*m)*lda+0+8*n]; 
    197. 

  197. 		 TempC[1] += TempA[3] * B[(3+8*m)*lda+1+8*n]; 
    198. 

  198. 		 TempC[2] += TempA[3] * B[(3+8*m)*lda+2+8*n]; 
    199. 

  199. 		 TempC[3] += TempA[3] * B[(3+8*m)*lda+3+8*n]; 
    200. 

  200. 		 TempC[4] += TempA[3] * B[(3+8*m)*lda+4+8*n]; 
    201. 

  201. 		 TempC[5] += TempA[3] * B[(3+8*m)*lda+5+8*n]; 
    202. 

  202. 		 TempC[6] += TempA[3] * B[(3+8*m)*lda+6+8*n]; 
    203. 

  203. 		 TempC[7] += TempA[3] * B[(3+8*m)*lda+7+8*n]; 
    204. 

  204. 

  205. 		 TempC[0] += TempA[4] * B[(4+8*m)*lda+0+8*n]; 
    206. 

  206. 		 TempC[1] += TempA[4] * B[(4+8*m)*lda+1+8*n]; 
    207. 

  207. 		 TempC[2] += TempA[4] * B[(4+8*m)*lda+2+8*n]; 
    208. 

  208. 		 TempC[3] += TempA[4] * B[(4+8*m)*lda+3+8*n]; 
    209. 

  209. 		 TempC[4] += TempA[4] * B[(4+8*m)*lda+4+8*n]; 
    210. 

  210. 		 TempC[5] += TempA[4] * B[(4+8*m)*lda+5+8*n]; 
    211. 

  211. 		 TempC[6] += TempA[4] * B[(4+8*m)*lda+6+8*n]; 
    212. 

  212. 		 TempC[7] += TempA[4] * B[(4+8*m)*lda+7+8*n]; 
    213. 

  213. 

  214. 

  215. 		 TempC[0] += TempA[5] * B[(5+8*m)*lda+0+8*n]; 
    216. 

  216. 		 TempC[1] += TempA[5] * B[(5+8*m)*lda+1+8*n]; 
    217. 

  217. 		 TempC[2] += TempA[5] * B[(5+8*m)*lda+2+8*n]; 
    218. 

  218. 		 TempC[3] += TempA[5] * B[(5+8*m)*lda+3+8*n]; 
    219. 

  219. 		 TempC[4] += TempA[5] * B[(5+8*m)*lda+4+8*n]; 
    220. 

  220. 		 TempC[5] += TempA[5] * B[(5+8*m)*lda+5+8*n]; 
    221. 

  221. 		 TempC[6] += TempA[5] * B[(5+8*m)*lda+6+8*n]; 
    222. 

  222. 		 TempC[7] += TempA[5] * B[(5+8*m)*lda+7+8*n]; 
    223. 

  223. 		 
    224. 

  224. 

  225. 

  226. 		 TempC[0] += TempA[6] * B[(6+8*m)*lda+0+8*n]; 
    227. 

  227. 		 TempC[1] += TempA[6] * B[(6+8*m)*lda+1+8*n]; 
    228. 

  228. 		 TempC[2] += TempA[6] * B[(6+8*m)*lda+2+8*n]; 
    229. 

  229. 		 TempC[3] += TempA[6] * B[(6+8*m)*lda+3+8*n]; 
    230. 

  230. 		 TempC[4] += TempA[6] * B[(6+8*m)*lda+4+8*n]; 
    231. 

  231. 		 TempC[5] += TempA[6] * B[(6+8*m)*lda+5+8*n]; 
    232. 

  232. 		 TempC[6] += TempA[6] * B[(6+8*m)*lda+6+8*n]; 
    233. 

  233. 		 TempC[7] += TempA[6] * B[(6+8*m)*lda+7+8*n]; 
    234. 

  234. 

  235. 

  236. 		 TempC[0] += TempA[7] * B[(7+8*m)*lda+0+8*n]; 
    237. 

  237. 		 TempC[1] += TempA[7] * B[(7+8*m)*lda+1+8*n]; 
    238. 

  238. 		 TempC[2] += TempA[7] * B[(7+8*m)*lda+2+8*n]; 
    239. 

  239. 		 TempC[3] += TempA[7] * B[(7+8*m)*lda+3+8*n]; 
    240. 

  240. 		 TempC[4] += TempA[7] * B[(7+8*m)*lda+4+8*n]; 
    241. 

  241. 		 TempC[5] += TempA[7] * B[(7+8*m)*lda+5+8*n]; 
    242. 

  242. 		 TempC[6] += TempA[7] * B[(7+8*m)*lda+6+8*n]; 
    243. 

  243. 		 TempC[7] += TempA[7] * B[(7+8*m)*lda+7+8*n]; 
    244. 

  244. 		 
    245. 

  245. 

  246. 		
    247. 

  247. 		 C[0+8*n+j*lda] += TempC[0];
    248. 

  248. 		 C[1+8*n+j*lda] += TempC[1];
    249. 

  249. 		 C[2+8*n+j*lda] += TempC[2];
    250. 

  250. 		 C[3+8*n+j*lda] += TempC[3];
    251. 

  251. 		 C[4+8*n+j*lda] += TempC[4];
    252. 

  252. 		 C[5+8*n+j*lda] += TempC[5];
    253. 

  253. 		 C[6+8*n+j*lda] += TempC[6];
    254. 

  254. 		 C[7+8*n+j*lda] += TempC[7];
    255. 

  255. 		 }
    256. 

  256.       }
    257. 

  257. 	 }
    258. 

  258. 	}
    259. 

  259. }
    260. 

  260.