eliminate_contiguous.hpp 598 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
#ifndef MIGRAPHX_GUARD_RTGLIB_ELIMINATE_CONTIGUOUS_HPP
#define MIGRAPHX_GUARD_RTGLIB_ELIMINATE_CONTIGUOUS_HPP
3
4

#include <string>
Paul's avatar
Paul committed
5
6
#include <migraphx/instruction_ref.hpp>
#include <migraphx/config.hpp>
7

Paul's avatar
Paul committed
8
namespace migraphx {
Paul's avatar
Paul committed
9
inline namespace MIGRAPHX_INLINE_NS {
10

Shucai Xiao's avatar
Shucai Xiao committed
11
struct module;
12

Paul's avatar
Paul committed
13
14
15
/**
 * Remove contiguous instructions by checking if the operator can use non-standard shapes.
 */
16
17
struct eliminate_contiguous
{
18
    std::string op_name;
19
    std::string name() const { return "eliminate_contiguous"; }
20
    void apply(module& m) const;
21
22
};

Paul's avatar
Paul committed
23
} // namespace MIGRAPHX_INLINE_NS
Paul's avatar
Paul committed
24
} // namespace migraphx
25
26

#endif