From fd62696bdbd42381fdd8a43ca1b20b9f278616b1 Mon Sep 17 00:00:00 2001
From: Russel Arbore <rarbore2@illinois.edu>
Date: Tue, 4 Mar 2025 13:12:28 -0600
Subject: [PATCH] "fix" backprop large bench on gpu
---
juno_samples/rodinia/backprop/benches/backprop_bench.rs | 2 +-
juno_samples/rodinia/backprop/src/gpu.sch | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/juno_samples/rodinia/backprop/benches/backprop_bench.rs b/juno_samples/rodinia/backprop/benches/backprop_bench.rs
index 492bce3b..80964c72 100644
--- a/juno_samples/rodinia/backprop/benches/backprop_bench.rs
+++ b/juno_samples/rodinia/backprop/benches/backprop_bench.rs
@@ -68,7 +68,7 @@ fn backprop_bench(c: &mut Criterion) {
};
bench("backprop bench small", 65536);
- bench("backprop bench large", 33554432);
+ bench("backprop bench large", 1048576);
}
criterion_group!(benches, backprop_bench);
diff --git a/juno_samples/rodinia/backprop/src/gpu.sch b/juno_samples/rodinia/backprop/src/gpu.sch
index f8cc84a3..cc9cc2ac 100644
--- a/juno_samples/rodinia/backprop/src/gpu.sch
+++ b/juno_samples/rodinia/backprop/src/gpu.sch
@@ -33,10 +33,10 @@ fixpoint {
reduce-slf(*);
simpl!(*);
-fork-extend[32](layer_forward@inner_loop);
+fork-extend[1024](layer_forward@inner_loop);
clean-monoid-reduces(layer_forward);
simpl!(layer_forward);
-fork-tile[32, 0, false, true](layer_forward@inner_loop);
+fork-tile[1024, 0, false, true](layer_forward@inner_loop);
clean-monoid-reduces(layer_forward);
let out = fork-split(layer_forward@inner_loop);
clean-monoid-reduces(layer_forward);
--
GitLab