target.hpp 377 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
#ifndef MIGRAPH_GUARD_MIGRAPHLIB_MIOPEN_TARGET_HPP
#define MIGRAPH_GUARD_MIGRAPHLIB_MIOPEN_TARGET_HPP
Paul's avatar
Paul committed
3

Paul's avatar
Paul committed
4
#include <migraph/program.hpp>
Paul's avatar
Paul committed
5

Paul's avatar
Paul committed
6
namespace migraph {
Paul's avatar
Paul committed
7
namespace gpu {
Paul's avatar
Paul committed
8

Paul's avatar
Paul committed
9
struct target
Paul's avatar
Paul committed
10
11
{
    std::string name() const;
Paul's avatar
Paul committed
12
    std::vector<pass> get_passes(migraph::context& gctx) const;
mei-ye's avatar
mei-ye committed
13
    migraph::context get_context() const;
Paul's avatar
Paul committed
14
};
Paul's avatar
Paul committed
15
} // namespace gpu
Paul's avatar
Paul committed
16
} // namespace migraph
Paul's avatar
Paul committed
17
18

#endif