Halide
AutoSchedule.h
Go to the documentation of this file.
1 #include "CostModel.h"
2 #include "Featurization.h"
3 #include "FunctionDAG.h"
4 #include "Halide.h"
5 #include "PerfectHashMap.h"
6 #include <vector>
7 
8 namespace Halide {
9 namespace Internal {
10 namespace Autoscheduler {
11 
13 
14 void find_and_apply_schedule(FunctionDAG &dag, const std::vector<Function> &outputs, const MachineParams &params,
15  CostModel *cost_model, int beam_size, StageMapOfScheduleFeatures *schedule_features);
16 
17 } // namespace Autoscheduler
18 } // namespace Internal
19 } // namespace Halide
Halide::MachineParams
A struct representing the machine parameters to generate the auto-scheduled code for.
Definition: Pipeline.h:31
Halide::Internal::Autoscheduler::FunctionDAG
Definition: FunctionDAG.h:346
CostModel.h
FunctionDAG.h
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition: AddAtomicMutex.h:21
PerfectHashMap.h
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::CostModel
Definition: CostModel.h:19
Halide::Internal::Autoscheduler::StageMapOfScheduleFeatures
PerfectHashMap< FunctionDAG::Node::Stage, ScheduleFeatures > StageMapOfScheduleFeatures
Definition: AutoSchedule.h:12
Halide::Internal::Autoscheduler::find_and_apply_schedule
void find_and_apply_schedule(FunctionDAG &dag, const std::vector< Function > &outputs, const MachineParams &params, CostModel *cost_model, int beam_size, StageMapOfScheduleFeatures *schedule_features)
Featurization.h
PerfectHashMap
Definition: PerfectHashMap.h:38