Function make_scan_slice_program

test/ref/scan_slice.cpp:40–60

}

static migraphx::program make_scan_slice_program(int64_t axis, int64_t direction)
{
    migraphx::program p;
    auto* mm = p.get_main_module();

    migraphx::shape data_sh{migraphx::shape::int32_type, {2, 2, 2}};
    std::vector<int> data(data_sh.elements());
    std::iota(data.begin(), data.end(), 0);
    auto data_lit = mm->add_literal(migraphx::literal{data_sh, data});

    migraphx::shape idx_sh{migraphx::shape::int64_type, {1}};
    auto idx_param = mm->add_parameter("idx", idx_sh);

    mm->add_instruction(migraphx::make_op("scan_slice", {{"axis", axis}, {"direction", direction}}),
                        data_lit,
                        idx_param);

    p.compile(migraphx::make_target("ref"));

    return p;
}

TEST_CASE(scan_slice_test_1)
{