Skip to content
Snippets Groups Projects
Commit 17d33842 authored by Aaron Councilman's avatar Aaron Councilman
Browse files

Unparallelize backprop

parent 8e3ea30c
No related branches found
No related tags found
2 merge requests!215Large benches,!214More optimizations
Pipeline #202026 passed
...@@ -30,27 +30,11 @@ simpl!(*); ...@@ -30,27 +30,11 @@ simpl!(*);
infer-schedules(*); 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(*); delete-uncalled(*);
const-inline(*); const-inline(*);
simpl!(*); simpl!(*);
fork-split(forward_body, adjust_body, output_error_body, hidden_error_body); fork-split(*);
unforkify(forward_body, adjust_body, output_error_body, hidden_error_body); unforkify(*);
gcm(*); gcm(*);
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment