Function TEST_CASE

test/gpu/literal.cpp:33–50

#include <migraphx/gpu/hip.hpp>

TEST_CASE(gpu_literal_test)
{
    migraphx::program p;
    auto* mm = p.get_main_module();
    auto lit = generate_literal(migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
    mm->add_literal(lit);
    p.compile(migraphx::make_target("gpu"));
    auto scratch = p.get_parameter("scratch");
    if(scratch == mm->end())
    {
        auto result = p.eval({}).back();
        EXPECT(lit == migraphx::gpu::from_gpu(result));
    }
    else
    {
        EXPECT(scratch->get_shape().bytes() == lit.get_shape().bytes());
    }
}

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