test_pad_highest.cpp 717 Bytes
Newer Older
1
2
3
4

#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
Paul's avatar
Paul committed
5
#include <migraphx/op/pad.hpp>
6
7
8
9
10
11

struct test_pad_highest : verify_program<test_pad_highest>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
12
        auto* mm = p.get_main_module();
13
14
15
        std::vector<migraphx::half> data0(4);
        std::iota(data0.begin(), data0.end(), 0);
        migraphx::shape s0{migraphx::shape::half_type, {2, 2}};
16
        auto l0 = mm->add_literal(migraphx::literal{s0, data0});
17
18
19
        migraphx::op::pad op{};
        op.value = std::numeric_limits<float>::max();
        op.pads  = {0, 0, 1, 1};
20
        mm->add_instruction(op, l0);
21
22
23
        return p;
    }
};