Commit e0100153 authored by Khalique's avatar Khalique
Browse files

added par_dfor similar to miopen test

parent e7cf8a26
...@@ -120,7 +120,7 @@ struct cpu_lrn ...@@ -120,7 +120,7 @@ struct cpu_lrn
float alphaoverarea = op.alpha / op.size; float alphaoverarea = op.alpha / op.size;
int radius = (op.size - 1) / 2; int radius = (op.size - 1) / 2;
dfor(n_batch, height, width)([&](int b, int h, int w) { par_dfor(n_batch, height, width)([&](int b, int h, int w) {
float scale = 0; float scale = 0;
dfor(channels)([&](int c) { dfor(channels)([&](int c) {
auto start = (c - radius) < 0 ? 0 : (c - radius); auto start = (c - radius) < 0 ? 0 : (c - radius);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment