feat: support 6 arguments

gorse-io · Aug 19, 2023 · 629600e · 629600e
1 parent 3ab80cb
commit 629600e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/tests/cs_npw.c → tests/l2.c b/tests/cs_npw.c → tests/l2.c
diff --git a/tests/mat_mul.c b/tests/mat_mul.c
@@ -0,0 +1,14 @@
+void mat_mul(float *a, float *b, float *res, long d1, long d2, long d3)
+{
+    // add the remaining vectors
+    for (int i = 0; i < d1; i++)
+    {
+        for (int j = 0; j < d3; j++) {
+            float sum = 0;
+            for (int k = 0; k < d2; k++) {
+                sum += a[i * d2 + k] * b[k * d3 + j];
+            }
+            res[i * d3 + j] = sum;
+        }
+    }
+}