Formatting

ae513aa8 · turneram · f89638ec · ae513aa8 · ae513aa8
Commit ae513aa8 authored May 02, 2022 by turneram
Show whitespace changes
Inline Side-by-side

Showing with 32 additions and 33 deletions

src/include/migraphx/op/layernorm.hpp src/include/migraphx/op/layernorm.hpp +30 -31

src/onnx/parse_layernorm.cpp src/onnx/parse_layernorm.cpp +2 -2

No files found.
--- a/src/include/migraphx/op/layernorm.hpp
+++ b/src/include/migraphx/op/layernorm.hpp
@@ -72,7 +72,7 @@ struct layernorm
                mean_inv_std_dev_dim.at(i) = 1;
        } */
-        if (args.size() == 3)
+        if(args.size() == 3)
        {
            visit_all(result, args[0], args[1], args[2])(
                [&](auto output, auto data, auto weights, auto bias) {
@@ -101,8 +101,7 @@ struct layernorm
        }
        else
        {
-            visit_all(result, args[0])(
+            visit_all(result, args[0])([&](auto output, auto data) {
-                [&](auto output, auto data) {
                par_for(norm_count, [&](auto idx) {
                    auto offset        = idx * norm_size;
                    double mean        = 0;

--- a/src/onnx/parse_layernorm.cpp
+++ b/src/onnx/parse_layernorm.cpp
@@ -32,7 +32,7 @@ struct parse_layernorm : op_parser<parse_layernorm>
        if(args.size() >= 2)
            layernorm = info.add_instruction(make_op("mul"), layernorm, args.at(1));
-        if (args.size() == 3)
+        if(args.size() == 3)
            layernorm = info.add_instruction(make_op("add"), layernorm, args.at(2));
        return layernorm;