Function TEST_CASE

test/fuse_concat.cpp:84–112

}

TEST_CASE(simple_concat_pointwise)
{
    migraphx::shape s{migraphx::shape::float_type, {2, 3}};
    migraphx::program p1;
    {
        auto* mm    = p1.get_main_module();
        auto x      = mm->add_parameter("x", s);
        auto y      = mm->add_parameter("y", s);
        auto add    = add_pointwise(p1, "main:pointwise0", {x, y}, single_pointwise("add"));
        auto sub    = add_pointwise(p1, "main:pointwise1", {x, y}, single_pointwise("sub"));
        auto concat = mm->add_instruction(migraphx::make_op("concat", {{"axis", 1}}), add, sub);
        mm->add_return({concat});
    }
    run_pass(p1);
    migraphx::program p2;
    {
        auto* mm = p2.get_main_module();
        auto x   = mm->add_parameter("x", s);
        auto y   = mm->add_parameter("y", s);
        auto fused_concat =
            add_pointwise_concat(p2,
                                 1,
                                 arg("noop:concat0", {}, noop_pointwise()),
                                 arg("concat:main:pointwise0", {x, y}, single_pointwise("add")),
                                 arg("concat:main:pointwise1", {x, y}, single_pointwise("sub")));
        mm->add_return({fused_concat});
    }
    EXPECT(p1 == p2);
}

TEST_CASE(partial_pointwise_concat)
{