Halide 22.0.0 Halide compiler and libraries

Loading...
Searching...
No Matches
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
12 typedef PerfectHashMap<FunctionDAG::Node::Stage, ScheduleFeatures> StageMapOfScheduleFeatures;
13
14 void find_and_apply_schedule(FunctionDAG &dag, const std::vector<Function> &outputs, const Adams2019Params &params,
15 CostModel *cost_model, int beam_size, StageMapOfScheduleFeatures *schedule_features);
16
17} // namespace Autoscheduler
18} // namespace Internal
19} // namespace Halide
PerfectHashMap.h
CostModel.h
Featurization.h
FunctionDAG.h
Halide::CostModel
Definition CostModel.h:61
PerfectHashMap
Definition PerfectHashMap.h:38
Halide::Internal::Autoscheduler::find_and_apply_schedule
void find_and_apply_schedule(FunctionDAG &dag, const std::vector< Function > &outputs, const Adams2019Params &params, CostModel *cost_model, int beam_size, StageMapOfScheduleFeatures *schedule_features)
Halide::Internal::Autoscheduler::StageMapOfScheduleFeatures
PerfectHashMap< FunctionDAG::Node::Stage, ScheduleFeatures > StageMapOfScheduleFeatures
Definition AutoSchedule.h:12
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition AbstractGenerator.h:19
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::cast
Internal::ConstantInterval cast(Type t, const Internal::ConstantInterval &a)
Cast operators for ConstantIntervals.
Halide::Internal::Autoscheduler::Adams2019Params
Definition CostModel.h:19
Halide::Internal::Autoscheduler::FunctionDAG
Definition FunctionDAG.h:367