From a8e41c3e7fb395e2f2d4a6b39832b08447f43d6c Mon Sep 17 00:00:00 2001
From: Russel Arbore <russel.jma@gmail.com>
Date: Sun, 2 Mar 2025 14:58:35 -0600
Subject: [PATCH] tweaks

---
 juno_samples/rodinia/srad/src/cpu.sch | 1 +
 juno_samples/rodinia/srad/src/gpu.sch | 1 +
 juno_samples/rodinia/srad/src/srad.jn | 8 ++++----
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/juno_samples/rodinia/srad/src/cpu.sch b/juno_samples/rodinia/srad/src/cpu.sch
index a4cd4956..44007fca 100644
--- a/juno_samples/rodinia/srad/src/cpu.sch
+++ b/juno_samples/rodinia/srad/src/cpu.sch
@@ -8,6 +8,7 @@ macro simpl!(X) {
   infer-schedules(X);
 }
 
+no-memset(srad@scratch);
 phi-elim(*);
 let loop1 = outline(srad@loop1);
 let loop2 = outline(srad@loop2);
diff --git a/juno_samples/rodinia/srad/src/gpu.sch b/juno_samples/rodinia/srad/src/gpu.sch
index f89b7ab8..c5a30527 100644
--- a/juno_samples/rodinia/srad/src/gpu.sch
+++ b/juno_samples/rodinia/srad/src/gpu.sch
@@ -8,6 +8,7 @@ macro simpl!(X) {
   infer-schedules(X);
 }
 
+no-memset(srad@scratch);
 phi-elim(*);
 let sum_loop = outline(srad@loop1);
 let main_loops = outline(srad@loop2 | srad@loop3);
diff --git a/juno_samples/rodinia/srad/src/srad.jn b/juno_samples/rodinia/srad/src/srad.jn
index 6074bf8c..176778be 100644
--- a/juno_samples/rodinia/srad/src/srad.jn
+++ b/juno_samples/rodinia/srad/src/srad.jn
@@ -50,10 +50,10 @@ fn srad<nrows, ncols: usize>(
     let varROI  = (sum2 / nelems as f32) - meanROI * meanROI;
     let q0sqr   = varROI / (meanROI * meanROI);
 
-    @dirs let dN : f32[ncols, nrows];
-    @dirs let dS : f32[ncols, nrows];
-    @dirs let dE : f32[ncols, nrows];
-    @dirs let dW : f32[ncols, nrows];
+    @scratch let dN : f32[ncols, nrows];
+    @scratch let dS : f32[ncols, nrows];
+    @scratch let dE : f32[ncols, nrows];
+    @scratch let dW : f32[ncols, nrows];
 
     let c : f32[ncols, nrows];
 
-- 
GitLab