pre_scheduling.cpp 341 Bytes
Newer Older
mei-ye's avatar
mei-ye committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#include <migraphx/pre_scheduling.hpp>
#include "pre_scheduling_impl.hpp"

namespace migraphx {

void pre_scheduling::apply(program& p) const
{
    if(!enabled(MIGRAPHX_DISABLE_PRE_SCHEDULING{}))
    {
        pre_scheduling_impl opt(&p, weight_func, num_of_streams, insert_instr, verify);
        opt.run();
    }
}
} // namespace migraphx