cpu_target.hpp 432 Bytes
Newer Older
mei-ye's avatar
mei-ye committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#ifndef MIGRAPH_GUARD_MIGRAPHLIB_CPU_TARGET_HPP
#define MIGRAPH_GUARD_MIGRAPHLIB_CPU_TARGET_HPP

#include <migraph/program.hpp>
#include <migraph/cpu/context.hpp>

namespace migraph {
namespace cpu {

struct cpu_target
{
    std::string name() const;
    std::vector<pass> get_passes(migraph::context& ctx) const;
    migraph::context get_context() const { return context{}; }
};

} // namespace cpu

} // namespace migraph

#endif