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