Function TEST_CASE

test/propagate_precision.cpp:43–77

}

TEST_CASE(propagate_input)
{
    migraphx::shape s1{migraphx::shape::half_type, {2, 3}};
    migraphx::shape s2{migraphx::shape::float_type, {2, 3}};
    migraphx::module m1;
    {
        auto x        = m1.add_parameter("x", s1);
        auto y        = m1.add_parameter("y", s2);
        auto two      = m1.add_literal(migraphx::literal{{migraphx::shape::half_type}, {2}});
        auto div      = migraphx::add_common_op(m1, migraphx::make_op("div"), {x, two});
        auto sqrt     = m1.add_instruction(migraphx::make_op("sqrt"), div);
        auto convert1 = m1.add_instruction(
            migraphx::make_op("convert", {{"target_type", migraphx::shape::float_type}}), sqrt);
        auto mul      = m1.add_instruction(migraphx::make_op("mul"), convert1, y);
        auto convert2 = m1.add_instruction(
            migraphx::make_op("convert", {{"target_type", migraphx::shape::half_type}}), mul);
        m1.add_return({convert2});
    }
    run_pass(m1);
    migraphx::module m2;
    {
        auto x        = m2.add_parameter("x", s1);
        auto y        = m2.add_parameter("y", s2);
        auto convert1 = m2.add_instruction(
            migraphx::make_op("convert", {{"target_type", migraphx::shape::float_type}}), x);
        auto two      = m2.add_literal(migraphx::literal{{migraphx::shape::half_type}, {2}});
        auto div      = migraphx::add_common_op(m2, migraphx::make_op("div"), {convert1, two});
        auto sqrt     = m2.add_instruction(migraphx::make_op("sqrt"), div);
        auto mul      = m2.add_instruction(migraphx::make_op("mul"), sqrt, y);
        auto convert2 = m2.add_instruction(
            migraphx::make_op("convert", {{"target_type", migraphx::shape::half_type}}), mul);
        m2.add_return({convert2});
    }
    EXPECT(m1.sort() == m2.sort());
}

TEST_CASE(propagate_output)
{