Halide
Halide::DefaultCostModel Class Reference

#include <DefaultCostModel.h>

Inherits Halide::CostModel.

Public Member Functions

 DefaultCostModel (const std::string &weights_in_path, const std::string &weights_out_path, bool randomize_weights)
 
 ~DefaultCostModel () override=default
 
void set_pipeline_features (const Internal::Autoscheduler::FunctionDAG &dag, const MachineParams &params) override
 
void set_pipeline_features (const Runtime::Buffer< float > &, int n)
 
void enqueue (const Internal::Autoscheduler::FunctionDAG &dag, const Halide::Internal::Autoscheduler::StageMapOfScheduleFeatures &schedule_feats, double *cost_ptr) override
 
void enqueue (int ns, Runtime::Buffer< float > *schedule_feats, double *cost_ptr)
 
void evaluate_costs () override
 
void reset () override
 
float backprop (const Runtime::Buffer< const float > &true_runtimes, float learning_rate)
 
void save_weights ()
 
void load_weights ()
 
- Public Member Functions inherited from Halide::CostModel
virtual ~CostModel ()=default
 

Detailed Description

Definition at line 10 of file DefaultCostModel.h.

Constructor & Destructor Documentation

◆ DefaultCostModel()

Halide::DefaultCostModel::DefaultCostModel ( const std::string &  weights_in_path,
const std::string &  weights_out_path,
bool  randomize_weights 
)
inline

Definition at line 27 of file DefaultCostModel.h.

References load_weights().

◆ ~DefaultCostModel()

Halide::DefaultCostModel::~DefaultCostModel ( )
overridedefault

Member Function Documentation

◆ set_pipeline_features() [1/2]

void Halide::DefaultCostModel::set_pipeline_features ( const Internal::Autoscheduler::FunctionDAG dag,
const MachineParams params 
)
overridevirtual

Implements Halide::CostModel.

◆ set_pipeline_features() [2/2]

void Halide::DefaultCostModel::set_pipeline_features ( const Runtime::Buffer< float > &  ,
int  n 
)

◆ enqueue() [1/2]

void Halide::DefaultCostModel::enqueue ( const Internal::Autoscheduler::FunctionDAG dag,
const Halide::Internal::Autoscheduler::StageMapOfScheduleFeatures schedule_feats,
double *  cost_ptr 
)
overridevirtual

Implements Halide::CostModel.

◆ enqueue() [2/2]

void Halide::DefaultCostModel::enqueue ( int  ns,
Runtime::Buffer< float > *  schedule_feats,
double *  cost_ptr 
)

◆ evaluate_costs()

void Halide::DefaultCostModel::evaluate_costs ( )
overridevirtual

Implements Halide::CostModel.

◆ reset()

void Halide::DefaultCostModel::reset ( )
overridevirtual

Implements Halide::CostModel.

◆ backprop()

float Halide::DefaultCostModel::backprop ( const Runtime::Buffer< const float > &  true_runtimes,
float  learning_rate 
)

◆ save_weights()

void Halide::DefaultCostModel::save_weights ( )

◆ load_weights()

void Halide::DefaultCostModel::load_weights ( )

Referenced by DefaultCostModel().


The documentation for this class was generated from the following file: