diff options
Diffstat (limited to 'examples/chill/gemm.c')
| -rw-r--r-- | examples/chill/gemm.c | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/examples/chill/gemm.c b/examples/chill/gemm.c index 355bafe..a565511 100644 --- a/examples/chill/gemm.c +++ b/examples/chill/gemm.c @@ -1,12 +1,15 @@ + +#define N 512 + int main() { - float a[512][512], b[512][512], c[512][512]; + float a[N][N], b[N][N], c[N][N]; int i, j, k; - int n; - for (j = 0; j < n; j++) - for (k = 0; k < n; k++) - for (i = 0; i < n; i++) { + + 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]; } |