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