From 17d338421b68e66c3c6e784a6a2e3ea901373383 Mon Sep 17 00:00:00 2001 From: Aaron Councilman <aaronjc4@illinois.edu> Date: Mon, 3 Mar 2025 16:11:19 -0600 Subject: [PATCH] Unparallelize backprop --- juno_samples/rodinia/backprop/src/cpu.sch | 20 ++------------------ 1 file changed, 2 insertions(+), 18 deletions(-) diff --git a/juno_samples/rodinia/backprop/src/cpu.sch b/juno_samples/rodinia/backprop/src/cpu.sch index 865cc1a2..6899523e 100644 --- a/juno_samples/rodinia/backprop/src/cpu.sch +++ b/juno_samples/rodinia/backprop/src/cpu.sch @@ -30,27 +30,11 @@ simpl!(*); infer-schedules(*); -fork-tile[32, 0, false, true](layer_forward@outer_loop \ layer_forward@inner_loop); -let (forward_outer, forward_inner) = fork-reshape[[1], [0]](layer_forward@outer_loop \ layer_forward@inner_loop); - -fork-tile[32, 0, false, true](adjust_weights); -let (adjust_outer, adjust_inner) = fork-reshape[[1], [0, 2]](adjust_weights); - -let forward_body = outline(forward_inner); -let adjust_body = outline(adjust_inner); - -rename["output_error"](output_error); -rename["hidden_error"](hidden_error); - -let output_error_body = auto-outline(output_error).output_error; -let hidden_error_body = auto-outline(hidden_error).hidden_error; - -inline(backprop); delete-uncalled(*); const-inline(*); simpl!(*); -fork-split(forward_body, adjust_body, output_error_body, hidden_error_body); -unforkify(forward_body, adjust_body, output_error_body, hidden_error_body); +fork-split(*); +unforkify(*); gcm(*); -- GitLab