Function TEST_CASE

test/gpu/codegen_literal.cpp:57–84

This test ensures that the codegen path doesn't round up literals, otherwise there are accuracy differences compared to ref. The values being passed in are 0.5 * (1/0.00787402), and after rounding must equal 63, not 64.

// The values being passed in are 0.5 * (1/0.00787402),
// and after rounding must equal 63, not 64.
TEST_CASE(mul_literal_round_test)
{

    migraphx::program p;
    auto* mm = p.get_main_module();
    migraphx::shape s0{migraphx::shape::float_type, {1}};
    auto l0 = mm->add_parameter("a", s0);
    auto l1 = mm->add_literal(1 / 0.00787402f);

    auto mul   = mm->add_instruction(migraphx::make_op("mul"), l0, l1);
    auto round = mm->add_instruction(migraphx::make_op("nearbyint"), mul);

    mm->add_return({round});

    migraphx::parameter_map m;
    std::vector<float> a = {0.5f};

    m["a"] = migraphx::argument{s0, a.data()};
    std::vector<float> ref_result;
    migraphx::target ref_t = migraphx::make_target("ref");
    run_prog(p, ref_t, m, ref_result);

    std::vector<float> gpu_result;
    migraphx::target gpu_t = migraphx::make_target("gpu");
    run_prog(p, gpu_t, m, gpu_result);

    EXPECT(migraphx::verify::verify_rms_range(gpu_result, ref_result));
}

int main(int argc, const char* argv[]) { test::run(argc, argv); }