Skip to content

GitLab

Commit 32e08558 authored by Umang Yadav's avatar Umang Yadav
Browse files

fix conv_bn eps

parent a8ef9124
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
test/verify/test_conv_bn.cpp
View file @ 32e08558
...@@ -59,7 +59,8 @@ struct test_conv_bn : verify_program<test_conv_bn<DType>> ...@@ -59,7 +59,8 @@ struct test_conv_bn : verify_program<test_conv_bn<DType>>
auto eps = mm->add_literal(migraphx::literal{DType, {1e-5f}}); auto eps = mm->add_literal(migraphx::literal{DType, {1e-5f}});
if constexpr((DType) == migraphx::shape::fp8e4m3fnuz_type) if constexpr((DType) == migraphx::shape::fp8e4m3fnuz_type)
{ {
eps = mm->add_literal(migraphx::literal{DType, {1e-1f}}); // use 1e-2f for the fp8
eps = mm->add_literal(migraphx::literal{DType, {1e-2f}});
} }
auto usq_scale = auto usq_scale =
......
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