added proper arguments to replace instruction

8ce572f8 · Aditya Atluri · c7debe34 · 8ce572f8
Commit 8ce572f8 authored Aug 07, 2018 by Aditya Atluri
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

src/targets/gpu/lowering.cpp src/targets/gpu/lowering.cpp +8 -2

No files found.
--- a/src/targets/gpu/lowering.cpp
+++ b/src/targets/gpu/lowering.cpp
@@ -388,8 +388,14 @@ struct miopen_apply
    {
        auto&& op   = any_cast<batch_norm_inference>(ins->op);
        auto output = insert_allocation(ins, ins->result);
-        prog->replace_instruction(
+        prog->replace_instruction(ins,
-            ins, miopen_batch_norm_inference{op}, ins->arguments.at(0), output);
+                                  miopen_batch_norm_inference{op},
+                                  ins->arguments.at(0),
+                                  ins->arguments.at(1),
+                                  ins->arguments.at(2),
+                                  ins->arguments.at(3),
+                                  ins->arguments.at(4),
+                                  output);
    }
 };