Function TEST_CASE

test/ref/add.cpp:34–55

#include <test.hpp>

TEST_CASE(add_broadcast_test)
{
    migraphx::program p;
    auto* mm = p.get_main_module();
    migraphx::shape a_shape{migraphx::shape::float_type, {2, 2, 3}};
    std::vector<float> a_data{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11};
    migraphx::shape b_shape{migraphx::shape::float_type, {2, 2}};
    std::vector<float> b_data{0, -1, -2, -3};
    uint64_t axis = 0;
    auto l1       = mm->add_literal(migraphx::literal{a_shape, a_data});
    auto l2       = mm->add_literal(migraphx::literal{b_shape, b_data});
    auto l3       = mm->add_instruction(
        migraphx::make_op("broadcast", {{"axis", axis}, {"out_lens", l1->get_shape().lens()}}), l2);
    mm->add_instruction(migraphx::make_op("add"), l1, l3);
    p.compile(migraphx::make_target("ref"));
    auto result = p.eval({}).back();
    EXPECT(result.get_shape().packed());
    std::vector<float> results_vector(12);
    result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
    std::vector<float> gold = {0, 1, 2, 2, 3, 4, 4, 5, 6, 6, 7, 8};
    EXPECT(migraphx::verify::verify_rms_range(results_vector, gold));
}

TEST_CASE(add_multibroadcast_test)
{