Function TEST_CASE

test/onnx/verify/resize_downsample_linear_dyn_test.cpp:29–51

#include <onnx_test.hpp>

TEST_CASE(resize_downsample_linear_dyn_test)
{
    using migraphx::half;
    migraphx::onnx_options options;
    options.map_dyn_input_dims = {{"X", {{1, 1}, {1, 1}, {2, 3}, {4, 8}}}};
    migraphx::program p        = read_onnx("resize_downsample_linear_half_test.onnx", options);
    p.compile(migraphx::make_target("ref"));

    migraphx::shape sx{migraphx::shape::half_type, {1, 1, 2, 4}};
    std::vector<half> dx = {half{1}, half{2}, half{3}, half{4}, half{5}, half{6}, half{7}, half{8}};

    migraphx::parameter_map pp;
    pp["X"] = migraphx::argument(sx, dx.data());

    auto result = p.eval(pp).back();
    std::vector<half> result_vector;
    result.visit([&](auto output) { result_vector.assign(output.begin(), output.end()); });

    // Expected output was calculated without any quantization
    std::vector<half> gold = {half{2.8333333}, half{4.833333}};

    EXPECT(migraphx::verify::verify_rms_range(result_vector, gold));
}