eliminate_allocation.hpp 533 Bytes
Newer Older
1
2
3
4
5
#ifndef MIGRAPH_GUARD_RTGLIB_ELIMINATE_ALLOCATION_HPP
#define MIGRAPH_GUARD_RTGLIB_ELIMINATE_ALLOCATION_HPP

#include <string>
#include <migraph/instruction_ref.hpp>
6
7
#include <migraph/config.hpp>

8
9
namespace migraph {
inline namespace MIGRAPH_INLINE_NS {
10
11
12
13
14

struct program;

struct eliminate_allocation
{
Paul's avatar
Paul committed
15
16
    std::string allocation_op{};
    std::size_t alignment = 32;
17
18
19
    std::string name() const { return "eliminate_allocation"; }
    void apply(program& p) const;
};
20

21
} // namespace MIGRAPH_INLINE_NS
22
23
24
} // namespace migraph

#endif