Skip to content
Snippets Groups Projects
Commit fd62696b authored by rarbore2's avatar rarbore2
Browse files

"fix" backprop large bench on gpu

parent 868d9729
No related branches found
No related tags found
2 merge requests!219Sequential schedules,!216More optimizations
Pipeline #202049 passed
This commit is part of merge request !216. Comments created here will be created in the context of that merge request.
...@@ -68,7 +68,7 @@ fn backprop_bench(c: &mut Criterion) { ...@@ -68,7 +68,7 @@ fn backprop_bench(c: &mut Criterion) {
}; };
bench("backprop bench small", 65536); bench("backprop bench small", 65536);
bench("backprop bench large", 33554432); bench("backprop bench large", 1048576);
} }
criterion_group!(benches, backprop_bench); criterion_group!(benches, backprop_bench);
......
...@@ -33,10 +33,10 @@ fixpoint { ...@@ -33,10 +33,10 @@ fixpoint {
reduce-slf(*); reduce-slf(*);
simpl!(*); simpl!(*);
fork-extend[32](layer_forward@inner_loop); fork-extend[1024](layer_forward@inner_loop);
clean-monoid-reduces(layer_forward); clean-monoid-reduces(layer_forward);
simpl!(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); clean-monoid-reduces(layer_forward);
let out = fork-split(layer_forward@inner_loop); let out = fork-split(layer_forward@inner_loop);
clean-monoid-reduces(layer_forward); clean-monoid-reduces(layer_forward);
......
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