From ded2703d52206ab24847140e9aa6a84c33862587 Mon Sep 17 00:00:00 2001 From: Aaron Councilman <aaronjc4@illinois.edu> Date: Wed, 12 Feb 2025 14:02:25 -0600 Subject: [PATCH] Make writes in cava medianMatrix simple --- juno_samples/cava/src/cava.jn | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/juno_samples/cava/src/cava.jn b/juno_samples/cava/src/cava.jn index 359a83ed..95a73f5b 100644 --- a/juno_samples/cava/src/cava.jn +++ b/juno_samples/cava/src/cava.jn @@ -2,10 +2,8 @@ fn medianMatrix<a : number, rows, cols : usize>(m : a[rows, cols]) -> a { const n : usize = rows * cols; let tmp : a[rows * cols]; - for r = 0 to rows { - for c = 0 to cols { - tmp[r * cols + c] = m[r, c]; - } + for i = 0 to rows * cols { + tmp[i] = m[i / cols, i % cols]; } for i = 0 to n - 1 { -- GitLab