Commit 20b9c7b0 authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

update eps value type

parent 5a068d7e
...@@ -58,7 +58,7 @@ __device__ void generic_binary_layernorm( ...@@ -58,7 +58,7 @@ __device__ void generic_binary_layernorm(
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);
auto eps_val = static_cast<value_type>(eps); value_type eps_val = eps; // implicit conversion for eps
r.inner([&](auto& y, auto x1, auto x2, auto... xs) { r.inner([&](auto& y, auto x1, auto x2, auto... xs) {
auto x = op(x1, x2); auto x = op(x1, x2);
......
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