19 #ifndef TVM_META_SCHEDULE_DATABASE_H_
20 #define TVM_META_SCHEDULE_DATABASE_H_
22 #include <tvm/ffi/container/array.h>
23 #include <tvm/ffi/function.h>
24 #include <tvm/ffi/reflection/registry.h>
25 #include <tvm/ffi/string.h>
37 namespace meta_schedule {
71 explicit Workload(ObjectPtr<WorkloadNode> data) : ObjectRef(data) {}
106 const ModuleEquality& mod_eq_;
128 refl::ObjectDef<TuningRecordNode>()
169 ffi::Optional<ffi::Array<FloatImm>> run_secs,
170 ffi::Optional<Target> target,
171 ffi::Optional<ffi::Array<ArgInfo>> args_info);
245 const ffi::String& workload_name);
254 const ffi::String& workload_name);
263 const ffi::String& workload_name);
280 std::unique_ptr<ModuleEquality> mod_eq_;
355 const ffi::String&)>;
360 using FSize = ffi::TypedFunction<int64_t()>;
397 ICHECK(
f_has_workload !=
nullptr) <<
"PyDatabase's HasWorkload method not implemented!";
402 ICHECK(
f_commit_workload !=
nullptr) <<
"PyDatabase's CommitWorkload method not implemented!";
408 <<
"PyDatabase's CommitTuningRecord method not implemented!";
413 ICHECK(
f_get_top_k !=
nullptr) <<
"PyDatabase's GetTopK method not implemented!";
419 <<
"PyDatabase's GetAllTuningRecords method not implemented!";
424 const ffi::String& workload_name)
final {
433 const ffi::String& workload_name)
final {
442 const ffi::String& workload_name)
final {
451 ICHECK(
f_size !=
nullptr) <<
"PyDatabase's Size method not implemented!";
469 explicit Database(ObjectPtr<DatabaseNode> data) : ObjectRef(data) {
470 TVM_FFI_ICHECK(data !=
nullptr);
484 ffi::String mod_eq_name =
"structural");
493 bool allow_missing, ffi::String mod_eq_name =
"structural");
533 ffi::String mod_eq_name =
"structural");
Managed reference class to IRModuleNode.
Definition: module.h:256
Managed reference class to TargetNode.
Definition: target.h:192
Managed reference to ScheduleNode.
Definition: schedule.h:885
Managed reference to TraceNode.
Definition: trace.h:143
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:308
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.
Compilation target object.