From 7c445a64d396f3e924659392deaf0854b5c255e9 Mon Sep 17 00:00:00 2001
From: Russel Arbore <russel.jma@gmail.com>
Date: Sun, 2 Mar 2025 15:29:54 -0600
Subject: [PATCH] parallelize loop2 in srad
---
juno_samples/rodinia/srad/src/cpu.sch | 9 +++++++--
juno_samples/rodinia/srad/src/srad.jn | 2 +-
2 files changed, 8 insertions(+), 3 deletions(-)
diff --git a/juno_samples/rodinia/srad/src/cpu.sch b/juno_samples/rodinia/srad/src/cpu.sch
index 5a8c180e..43d8ceac 100644
--- a/juno_samples/rodinia/srad/src/cpu.sch
+++ b/juno_samples/rodinia/srad/src/cpu.sch
@@ -35,7 +35,12 @@ simpl!(*);
slf(*);
simpl!(*);
-fork-split(*);
-unforkify(*);
+fork-tile[32, 0, false, false](loop2);
+let split = fork-split(loop2);
+let loop2_body = outline(split.srad_1.fj1);
+simpl!(loop2, loop2_body);
+
+fork-split(extract, compress, loop1, loop2_body, loop3);
+unforkify(extract, compress, loop1, loop2_body, loop3);
gcm(*);
diff --git a/juno_samples/rodinia/srad/src/srad.jn b/juno_samples/rodinia/srad/src/srad.jn
index b055b296..024be598 100644
--- a/juno_samples/rodinia/srad/src/srad.jn
+++ b/juno_samples/rodinia/srad/src/srad.jn
@@ -51,7 +51,7 @@ fn srad<nrows, ncols: usize>(
@scratch let dE : f32[ncols, nrows];
@scratch let dW : f32[ncols, nrows];
- let c : f32[ncols, nrows];
+ @scratch let c : f32[ncols, nrows];
@loop2 for j in 0..ncols {
for i in 0..nrows {
--
GitLab