Function TEST_CASE

test/split_reduce.cpp:56–76

}

TEST_CASE(single)
{
    migraphx::shape s{migraphx::shape::float_type, {2, 3, 327680}};
    migraphx::program p1;
    {
        auto* mm  = p1.get_main_module();
        auto x    = mm->add_parameter("x", s);
        auto rsum = mm->add_instruction(migraphx::make_op("reduce_sum", {{"axes", {2}}}), x);
        mm->add_return({rsum});
    }
    run_pass(p1);
    migraphx::program p2;
    {
        auto* mm  = p2.get_main_module();
        auto x    = mm->add_parameter("x", s);
        auto rsum = add_reduce(
            p2, "main:reduce_sum0_split", {x}, {2}, "assign_add", single_reduce("reduce_sum"));
        mm->add_return({rsum});
    }
    EXPECT(p1 == p2);
}

TEST_CASE(fused)
{