19 #ifndef TVM_META_SCHEDULE_TUNE_CONTEXT_H_
20 #define TVM_META_SCHEDULE_TUNE_CONTEXT_H_
22 #include <tvm/ffi/container/array.h>
23 #include <tvm/ffi/container/map.h>
24 #include <tvm/ffi/function.h>
25 #include <tvm/ffi/optional.h>
26 #include <tvm/ffi/reflection/registry.h>
27 #include <tvm/ffi/string.h>
39 namespace meta_schedule {
41 class TaskSchedulerNode;
42 class MeasureCallback;
51 Optional<IRModule>
mod;
69 refl::ObjectDef<TuneContextNode>()
90 static constexpr
const char*
_type_key =
"meta_schedule.TuneContext";
113 Optional<SpaceGenerator> space_generator,
114 Optional<SearchStrategy> search_strategy, Optional<String> task_name,
The auto tuning context.
Definition: tune_context.h:46
static constexpr const char * _type_key
Definition: tune_context.h:90
TRandState rand_state
The random state.
Definition: tune_context.h:63
static void RegisterReflection()
Definition: tune_context.h:67
Optional< IRModule > mod
The workload to be tuned.
Definition: tune_context.h:51
Optional< String > task_name
The name of the tuning task.
Definition: tune_context.h:59
TuneContext Clone() const
Clone the tune context.
Optional< SpaceGenerator > space_generator
The design space generator.
Definition: tune_context.h:55
void Initialize()
Initialize members that needs initialization with tune context.
TVM_DECLARE_FINAL_OBJECT_INFO(TuneContextNode, Object)
support::LinearCongruentialEngine::TRandState TRandState
Definition: tune_context.h:48
int num_threads
The number of threads to be used.
Definition: tune_context.h:61
ffi::Function logger
The tuning task's logging function. t.
Definition: tune_context.h:65
Optional< SearchStrategy > search_strategy
The search strategy.
Definition: tune_context.h:57
Optional< Target > target
The target to be tuned for.
Definition: tune_context.h:53
Managed reference to TuneContextNode.
Definition: tune_context.h:98
TuneContext(Optional< IRModule > mod, Optional< Target > target, Optional< SpaceGenerator > space_generator, Optional< SearchStrategy > search_strategy, Optional< String > task_name, int num_threads, TRandState rand_state, ffi::Function logger)
Constructor.
TVM_DEFINE_MUTABLE_NOTNULLABLE_OBJECT_REF_METHODS(TuneContext, ObjectRef, TuneContextNode)
support::LinearCongruentialEngine::TRandState TRandState
Definition: tune_context.h:100
int64_t TRandState
Definition: random_engine.h:46
IRModule that holds the functions and type definitions.
Definition: repr_printer.h:91
tvm::PrimExpr mod(const tvm::PrimExpr &a, const tvm::PrimExpr &b)
Definition: broadcast.h:306
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.
Random number generator. It provides a generic interface consistent with std::uniform_random_bit_gene...
Compilation target object.