20 #ifndef TVM_META_SCHEDULE_FEATURE_EXTRACTOR_H_
21 #define TVM_META_SCHEDULE_FEATURE_EXTRACTOR_H_
23 #include <tvm/ffi/container/array.h>
24 #include <tvm/ffi/function.h>
25 #include <tvm/ffi/reflection/registry.h>
26 #include <tvm/ffi/string.h>
32 namespace meta_schedule {
53 const TuneContext& context,
const ffi::Array<MeasureCandidate>& candidates) = 0;
66 using FExtractFrom = ffi::TypedFunction<ffi::Array<tvm::runtime::Tensor>(
67 const TuneContext& context,
const ffi::Array<MeasureCandidate>& candidates)>;
72 using FAsString = ffi::TypedFunction<ffi::String()>;
85 const TuneContext& context,
const ffi::Array<MeasureCandidate>& candidates)
final;
107 int arith_intensity_curve_num_samples = 10,
108 int cache_line_bytes = 64,
109 bool extract_workload =
false);
Managed reference to TuneContextNode.
Definition: tune_context.h:98
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.
A device-independent managed Tensor abstraction.