Function TEST_CASE

test/ref/pack_unpack_fp4.cpp:34–50

#include <test.hpp>

TEST_CASE(pack_fp4)
{
    migraphx::program p;
    auto* mm = p.get_main_module();
    migraphx::shape s{migraphx::shape::float_type, {2, 2}};
    auto l0 = mm->add_literal(migraphx::literal{s, {-2.f, 3.4f, 3.5f, 0.f}});
    mm->add_instruction(migraphx::make_op("pack_fp4"), l0);
    p.compile(migraphx::make_target("ref"));
    auto result = p.eval({}).back();
    result =
        result.reshape(migraphx::shape(migraphx::shape::uint8_type, result.get_shape().lens()));
    std::vector<uint8_t> results_vector(2);
    result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
    std::vector<uint8_t> gold{0x5C, 0x06};
    EXPECT(results_vector.at(0) == gold.at(0));
    EXPECT(results_vector.at(1) == gold.at(1));
}

TEST_CASE(unpack_fp4)
{