Skip to content

GitLab

Commit cc3d899c authored by Ahsan Saghir's avatar Ahsan Saghir
Browse files

Change shape for one of the tests

parent 062e7554
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 9 deletions
test/autocast_fp8.cpp
View file @ cc3d899c
......@@ -120,10 +120,10 @@ TEST_CASE(autocast_fp8_4)
{
migraphx::module m1;
{
auto x1 = m1.add_parameter("x1", {migraphx::shape::fp8e4m3fnuz_type, {1}});
auto y1 = m1.add_parameter("y1", {migraphx::shape::fp8e4m3fnuz_type, {1}});
auto x2 = m1.add_parameter("x2", {migraphx::shape::float_type, {1}});
auto y2 = m1.add_parameter("y2", {migraphx::shape::float_type, {1}});
auto x1 = m1.add_parameter("x1", {migraphx::shape::fp8e4m3fnuz_type, {3, 4}, {1, 3}});
auto y1 = m1.add_parameter("y1", {migraphx::shape::fp8e4m3fnuz_type, {3, 4}, {1, 3}});
auto x2 = m1.add_parameter("x2", {migraphx::shape::float_type, {3, 4}, {1, 3}});
auto y2 = m1.add_parameter("y2", {migraphx::shape::float_type, {3, 4}, {1, 3}});
auto sum1 = m1.add_instruction(migraphx::make_op("add"), x1, y1);
auto sum2 = m1.add_instruction(migraphx::make_op("add"), x2, y2);
m1.add_return({sum1, sum2});
......@@ -132,10 +132,10 @@ TEST_CASE(autocast_fp8_4)
migraphx::module m2;
{
auto x2 = m2.add_parameter("x2", {migraphx::shape::float_type, {1}});
auto y2 = m2.add_parameter("y2", {migraphx::shape::float_type, {1}});
auto y1 = m2.add_parameter("y1", {migraphx::shape::float_type, {1}});
auto x1 = m2.add_parameter("x1", {migraphx::shape::float_type, {1}});
auto x2 = m2.add_parameter("x2", {migraphx::shape::float_type, {3, 4}, {1, 3}});
auto y2 = m2.add_parameter("y2", {migraphx::shape::float_type, {3, 4}, {1, 3}});
auto y1 = m2.add_parameter("y1", {migraphx::shape::float_type, {3, 4}, {1, 3}});
auto x1 = m2.add_parameter("x1", {migraphx::shape::float_type, {3, 4}, {1, 3}});
auto y1_fp8 = m2.add_instruction(migraphx::make_op("convert", {{"target_type", migraphx::shape::fp8e4m3fnuz_type}}), y1);
auto x1_fp8 = m2.add_instruction(migraphx::make_op("convert", {{"target_type", migraphx::shape::fp8e4m3fnuz_type}}), x1);
auto sum1 = m2.add_instruction(migraphx::make_op("add"), x1_fp8, y1_fp8);
......@@ -164,7 +164,6 @@ TEST_CASE(autocast_fp8_5)
auto y = m2.add_parameter("y", {migraphx::shape::float_type, {1}});
auto sum = m2.add_instruction(migraphx::make_op("add"), x, y);
m2.add_return({sum});
}
EXPECT(m1 == m2);
}
......
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