summaryrefslogtreecommitdiff
path: root/test-chill/test-cases/examples/chill/gemm.c
diff options
context:
space:
mode:
Diffstat (limited to 'test-chill/test-cases/examples/chill/gemm.c')
-rw-r--r--test-chill/test-cases/examples/chill/gemm.c25
1 files changed, 25 insertions, 0 deletions
diff --git a/test-chill/test-cases/examples/chill/gemm.c b/test-chill/test-cases/examples/chill/gemm.c
new file mode 100644
index 0000000..2c90ea5
--- /dev/null
+++ b/test-chill/test-cases/examples/chill/gemm.c
@@ -0,0 +1,25 @@
+
+#ifndef N
+#define N 512
+#endif
+
+/*
+<test name=gemm define="{'N':512}">
+procedure int gemm(
+ in float[N][N] a = matrix([,], lambda i,j: random(2,-2)),
+ in float[N][N] b = matrix([,], lambda i,j: random(2,-2)),
+ out float[N][N] c = matrix([,], lambda i,j: 0))
+</test>
+*/
+int gemm(float a[N][N], float b[N][N], float c[N][N]) {
+ int i, j, k;
+ int n = N;
+ for (j = 0; j < n; j++)
+ for (k = 0; k < n; k++)
+ for (i = 0; i < n; i++) {
+ c[i][j] = c[i][j] + a[i][k] * b[k][j];
+ }
+
+ return 0;
+}
+