Commit 8e015bd0 authored by mei-ye's avatar mei-ye
Browse files

more format changes and add header file to event.hpp

parent 9d87cd50
...@@ -12,7 +12,7 @@ struct program_visitor ...@@ -12,7 +12,7 @@ struct program_visitor
instruction_ref end() { return reversed ? p_program->begin() : std::prev(p_program->end()); } instruction_ref end() { return reversed ? p_program->begin() : std::prev(p_program->end()); }
instruction_ref next(instruction_ref ins) { return reversed ? std::prev(ins) : std::next(ins); } instruction_ref next(instruction_ref ins) { return reversed ? std::prev(ins) : std::next(ins); }
const std::vector<instruction_ref>& get_inputs(instruction_ref ins) const std::vector<instruction_ref>& get_inputs(instruction_ref ins) const
{ {
return reversed ? ins->outputs() : ins->inputs(); return reversed ? ins->outputs() : ins->inputs();
} }
......
...@@ -122,7 +122,7 @@ void pre_scheduling_impl::reorder() ...@@ -122,7 +122,7 @@ void pre_scheduling_impl::reorder()
// Assign stream to nodes according to load balance. // Assign stream to nodes according to load balance.
// //
int pre_scheduling_impl::get_stream(stream_info& info, dag_node* node) int pre_scheduling_impl::get_stream(stream_info& info, dag_node* node) const
{ {
int max_cycle = info.max_cycle; int max_cycle = info.max_cycle;
if(max_cycle == 0) if(max_cycle == 0)
......
...@@ -104,7 +104,7 @@ struct pre_scheduling_impl ...@@ -104,7 +104,7 @@ struct pre_scheduling_impl
} }
void schedule(std::list<dag_node*>&); void schedule(std::list<dag_node*>&);
void compute_weights(); void compute_weights();
int get_stream(stream_info&, dag_node*); int get_stream(stream_info&, dag_node*) const;
void record(stream_info&, dag_node*); void record(stream_info&, dag_node*);
void reorder(); void reorder();
void run(); void run();
......
...@@ -4,6 +4,7 @@ ...@@ -4,6 +4,7 @@
#include <migraphx/instruction.hpp> #include <migraphx/instruction.hpp>
#include <migraphx/operators.hpp> #include <migraphx/operators.hpp>
#include <migraphx/config.hpp> #include <migraphx/config.hpp>
#include <migraphx/gpu/context.hpp>
#include <utility> #include <utility>
namespace migraphx { namespace migraphx {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment