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