1 #ifndef AUTO_SCHEDULE_H 2 #define AUTO_SCHEDULE_H 17 namespace Autoscheduler {
22 const std::vector<Function> &outputs,
23 const Anderson2021Params ¶ms,
25 CostModel *cost_model,
33 #endif // AUTO_SCHEDULE_H
This file defines the class FunctionDAG, which is our representation of a Halide pipeline, and contains methods to using Halide's bounds tools to query properties of it.
Not visible externally, similar to 'static' linkage in C.
PerfectHashMap< FunctionDAG::Node::Stage, ScheduleFeatures > StageMapOfScheduleFeatures
void find_and_apply_schedule(FunctionDAG &dag, const std::vector< Function > &outputs, const Adams2019Params ¶ms, CostModel *cost_model, int beam_size, StageMapOfScheduleFeatures *schedule_features)