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(
auto mean_x = means[0];
auto mean_x2 = means[1];
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) {
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