tvm
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::s_tir::meta_schedule::ExtractedTaskObj Class Reference

A tuning task extracted from the high-level IR. More...

#include <extracted_task.h>

Inheritance diagram for tvm::s_tir::meta_schedule::ExtractedTaskObj:
Collaboration diagram for tvm::s_tir::meta_schedule::ExtractedTaskObj:

Public Member Functions

 ExtractedTaskObj (ffi::UnsafeInit tag)
 
 ExtractedTaskObj (Target target)
 
 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.meta_schedule.ExtractedTask", ExtractedTaskObj, ffi::Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

ffi::String task_name
 The name of the task extracted.
 
IRModule mod
 The high-level IR.
 
Target target
 Target.
 
ffi::Array< IRModule > dispatched
 A list of low-level IRs that the high-level IR could potentially dispatch to.
 
int weight
 Weight of the task.
 

Static Public Attributes

static constexpr const bool _type_mutable = true
 

Detailed Description

A tuning task extracted from the high-level IR.

Constructor & Destructor Documentation

◆ ExtractedTaskObj() [1/2]

tvm::s_tir::meta_schedule::ExtractedTaskObj::ExtractedTaskObj ( ffi::UnsafeInit  tag)
inlineexplicit

◆ ExtractedTaskObj() [2/2]

tvm::s_tir::meta_schedule::ExtractedTaskObj::ExtractedTaskObj ( Target  target)
inlineexplicit

Member Function Documentation

◆ RegisterReflection()

static void tvm::s_tir::meta_schedule::ExtractedTaskObj::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::meta_schedule::ExtractedTaskObj::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "s_tir.meta_schedule.ExtractedTask"  ,
ExtractedTaskObj  ,
ffi::Object   
)

Member Data Documentation

◆ _type_mutable

constexpr const bool tvm::s_tir::meta_schedule::ExtractedTaskObj::_type_mutable = true
staticconstexpr

◆ dispatched

ffi::Array<IRModule> tvm::s_tir::meta_schedule::ExtractedTaskObj::dispatched

A list of low-level IRs that the high-level IR could potentially dispatch to.

◆ mod

IRModule tvm::s_tir::meta_schedule::ExtractedTaskObj::mod

The high-level IR.

◆ target

Target tvm::s_tir::meta_schedule::ExtractedTaskObj::target

◆ task_name

ffi::String tvm::s_tir::meta_schedule::ExtractedTaskObj::task_name

The name of the task extracted.

◆ weight

int tvm::s_tir::meta_schedule::ExtractedTaskObj::weight

Weight of the task.


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