"vscode:/vscode.git/clone" did not exist on "865d838dae4bb03a06eb07d0e0f303de5c11dfa1"
literal.cpp 737 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
#include <test.hpp>
#include <basic_ops.hpp>
Paul's avatar
Paul committed
3
4
5
6
7
#include <migraphx/program.hpp>
#include <migraphx/instruction.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/gpu/target.hpp>
#include <migraphx/gpu/hip.hpp>
Paul's avatar
Paul committed
8

Paul's avatar
Paul committed
9
void gpu_literal_test()
Paul's avatar
Paul committed
10
{
Paul's avatar
Paul committed
11
12
    migraphx::program p;
    auto lit = generate_literal(migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
Paul's avatar
Paul committed
13
    p.add_literal(lit);
Paul's avatar
Paul committed
14
    p.compile(migraphx::gpu::target{});
15
    auto scratch = p.get_parameter("scratch");
Paul's avatar
Paul committed
16
17
    if(scratch == p.end())
    {
18
        auto result = p.eval({}).back();
Paul's avatar
Paul committed
19
        EXPECT(lit == migraphx::gpu::from_gpu(result));
Paul's avatar
Paul committed
20
21
22
    }
    else
    {
23
24
        EXPECT(scratch->get_shape().bytes() == lit.get_shape().bytes());
    }
Paul's avatar
Paul committed
25
26
}

Paul's avatar
Paul committed
27
int main() { gpu_literal_test(); }