target.hpp 478 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>
5
#include <migraph/config.hpp>
Paul's avatar
Paul committed
6

7
namespace migraph { inline namespace MIGRAPH_INLINE_NS {
Paul's avatar
Paul committed
8
namespace gpu {
Paul's avatar
Paul committed
9

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

Paul's avatar
Paul committed
17
} // namespace gpu
18
} // namespace MIGRAPH_INLINE_NS
Paul's avatar
Paul committed
19
} // namespace migraph
Paul's avatar
Paul committed
20
21

#endif