Skip to content

GitLab

Commit ed34d352 authored by Paul's avatar Paul
Browse files

Format

parent 3143befc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/targets/gpu/kernels/include/migraphx/kernels/layernorm.hpp
View file @ ed34d352
...@@ -55,8 +55,8 @@ __device__ void generic_binary_layernorm( ...@@ -55,8 +55,8 @@ __device__ void generic_binary_layernorm(
return make_array(x, x * x) * vec_type<value_type>{1.0 / relements}; return make_array(x, x * x) * vec_type<value_type>{1.0 / relements};
})(input1, input2); })(input1, input2);
auto mean_x = means[0]; auto mean_x = means[0];
auto mean_x2 = means[1]; auto mean_x2 = means[1];
auto variance = mean_x2 - (mean_x * mean_x); auto variance = mean_x2 - (mean_x * mean_x);
r.inner([&](auto& y, auto x1, auto x2, auto... xs) { r.inner([&](auto& y, auto x1, auto x2, auto... xs) {
......
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