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 Array<MeasureCandidate>& candidates) = 0;
55 static constexpr
const char*
_type_key =
"meta_schedule.FeatureExtractor";
68 using FExtractFrom = ffi::TypedFunction<Array<tvm::runtime::NDArray>(
69 const TuneContext& context,
const Array<MeasureCandidate>& candidates)>;
87 const Array<MeasureCandidate>& candidates)
final;
89 static constexpr
const char*
_type_key =
"meta_schedule.PyFeatureExtractor";
110 int arith_intensity_curve_num_samples = 10,
111 int cache_line_bytes = 64,
112 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 device-independent managed NDArray abstraction.
A managed object in the TVM runtime.