#include <database.h>

Inheritance diagram for tvm::s_tir::meta_schedule::DatabaseNode:

Collaboration diagram for tvm::s_tir::meta_schedule::DatabaseNode:

Public Member Functions
	DatabaseNode (ffi::String mod_eq_name="structural")
	Constructor.

virtual	~DatabaseNode ()
	Default destructor.

virtual bool	HasWorkload (const IRModule &mod)=0
	Check if the database has the given workload.

virtual Workload	CommitWorkload (const IRModule &mod)=0
	Look up or add workload to the database if missing.

virtual void	CommitTuningRecord (const TuningRecord &record)=0
	Add a tuning record to the database.

virtual ffi::Array< TuningRecord >	GetTopK (const Workload &workload, int top_k)=0
	Get the top K valid tuning records of given workload from the database.

virtual ffi::Array< TuningRecord >	GetAllTuningRecords ()=0
	Get all tuning records from the database.

virtual int64_t	Size ()=0
	Get the size of the database.

virtual ffi::Optional< TuningRecord >	QueryTuningRecord (const IRModule &mod, const Target &target, const ffi::String &workload_name)
	Query the best record of the given workload from the database.

virtual ffi::Optional< s_tir::Schedule >	QuerySchedule (const IRModule &mod, const Target &target, const ffi::String &workload_name)
	Query the best schedule of the given workload from the database.

virtual ffi::Optional< IRModule >	QueryIRModule (const IRModule &mod, const Target &target, const ffi::String &workload_name)
	Query the best IRModule of the given workload from the database.

void	DumpPruned (Database destination)
	Prune the database and dump it a given database.

const ModuleEquality &	GetModuleEquality () const
	Return a reference to the owned module equality method instance.

	TVM_FFI_DECLARE_OBJECT_INFO ("s_tir.meta_schedule.Database", DatabaseNode, ffi::Object)

Static Public Attributes
static constexpr const bool	_type_mutable = true

Constructor & Destructor Documentation

◆ DatabaseNode()

tvm::s_tir::meta_schedule::DatabaseNode::DatabaseNode ( ffi::String mod_eq_name = "structural" )

explicit

Constructor.

Parameters

mod_eq_name

A string to specify the module equality testing and hashing method. It must be one of the followings:

"structural": Use StructuralEqual/Hash
"ignore-tensor": Same as "structural", but ignore tensor raw data during equality testing and hashing.
"anchor-block": Apply equality testing and hashing on the anchor block extracted from a given module. The "ignore-tensor" varint is used for the extracted blocks or in case no anchor block is found. For the definition of the anchor block, see tvm/tirx/analysis.h.

◆ ~DatabaseNode()

virtual tvm::s_tir::meta_schedule::DatabaseNode::~DatabaseNode ( )

virtual

Default destructor.

Member Function Documentation

◆ CommitTuningRecord()

virtual void tvm::s_tir::meta_schedule::DatabaseNode::CommitTuningRecord ( const TuningRecord & record )

pure virtual

Add a tuning record to the database.

Parameters

record The tuning record to be added.

Implemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ CommitWorkload()

virtual Workload tvm::s_tir::meta_schedule::DatabaseNode::CommitWorkload ( const IRModule & mod )

pure virtual

Look up or add workload to the database if missing.

Parameters

mod	The IRModule to be searched for or added.

Returns: The workload corresponding to the given IRModule.

Implemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ DumpPruned()

void tvm::s_tir::meta_schedule::DatabaseNode::DumpPruned ( Database destination )

Prune the database and dump it a given database.

Parameters

destination The destination database to be dumped to.

◆ GetAllTuningRecords()

virtual ffi::Array< TuningRecord > tvm::s_tir::meta_schedule::DatabaseNode::GetAllTuningRecords ( )

pure virtual

Get all tuning records from the database.

Returns: An Array of all the tuning records in the database.

Implemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ GetModuleEquality()

const ModuleEquality & tvm::s_tir::meta_schedule::DatabaseNode::GetModuleEquality ( ) const

inline

Return a reference to the owned module equality method instance.

◆ GetTopK()

virtual ffi::Array< TuningRecord > tvm::s_tir::meta_schedule::DatabaseNode::GetTopK	(	const Workload &	workload,
		int	top_k
	)

pure virtual

Get the top K valid tuning records of given workload from the database.

Parameters

workload	The workload to be searched for.
top_k	The number of top records to be returned.

Returns: An array of top K tuning records for the given workload.

Implemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ HasWorkload()

virtual bool tvm::s_tir::meta_schedule::DatabaseNode::HasWorkload ( const IRModule & mod )

pure virtual

Check if the database has the given workload.

Parameters

mod	The IRModule to be searched for.

Returns: Whether the database has the given workload.

Implemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ QueryIRModule()

virtual ffi::Optional< IRModule > tvm::s_tir::meta_schedule::DatabaseNode::QueryIRModule	(	const IRModule &	mod,
		const Target &	target,
		const ffi::String &	workload_name
	)

virtual

Query the best IRModule of the given workload from the database.

Parameters

mod	The IRModule to be searched for.
target	The target to be searched for.
workload_name	The name of the workload to be searched for.

Returns: The IRModule in the best IRModule of the given workload; std::nullopt if not found.

Reimplemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ QuerySchedule()

virtual ffi::Optional< s_tir::Schedule > tvm::s_tir::meta_schedule::DatabaseNode::QuerySchedule	(	const IRModule &	mod,
		const Target &	target,
		const ffi::String &	workload_name
	)

virtual

Query the best schedule of the given workload from the database.

Parameters

mod	The IRModule to be searched for.
target	The target to be searched for.
workload_name	The name of the workload to be searched for.

Returns: The schedule in the best schedule of the given workload; std::nullopt if not found.

Reimplemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ QueryTuningRecord()

virtual ffi::Optional< TuningRecord > tvm::s_tir::meta_schedule::DatabaseNode::QueryTuningRecord	(	const IRModule &	mod,
		const Target &	target,
		const ffi::String &	workload_name
	)

virtual

Query the best record of the given workload from the database.

Parameters

mod	The IRModule to be searched for.
target	The target to be searched for.
workload_name	The name of the workload to be searched for.

Returns: The best record of the given workload; std::nullopt if not found.

Reimplemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ Size()

virtual int64_t tvm::s_tir::meta_schedule::DatabaseNode::Size ( )

pure virtual

Get the size of the database.

Returns: The size of the database.

Implemented in tvm::s_tir::meta_schedule::PyDatabaseNode.

◆ TVM_FFI_DECLARE_OBJECT_INFO()

tvm::s_tir::meta_schedule::DatabaseNode::TVM_FFI_DECLARE_OBJECT_INFO	(	"s_tir.meta_schedule.Database"	,
		DatabaseNode	,
		ffi::Object
	)

Member Data Documentation

◆ _type_mutable

constexpr const bool tvm::s_tir::meta_schedule::DatabaseNode::_type_mutable = true

staticconstexpr

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

include/tvm/s_tir/meta_schedule/database.h

Public Member Functions

Static Public Attributes

Constructor & Destructor Documentation

◆ DatabaseNode()

◆ ~DatabaseNode()

Member Function Documentation

◆ CommitTuningRecord()

◆ CommitWorkload()

◆ DumpPruned()

◆ GetAllTuningRecords()

◆ GetModuleEquality()

◆ GetTopK()

◆ HasWorkload()

◆ QueryIRModule()

◆ QuerySchedule()

◆ QueryTuningRecord()

◆ Size()

◆ TVM_FFI_DECLARE_OBJECT_INFO()

Member Data Documentation

◆ _type_mutable