19 #ifndef TVM_META_SCHEDULE_BUILDER_H_
20 #define TVM_META_SCHEDULE_BUILDER_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>
34 namespace meta_schedule {
44 ffi::Optional<ffi::Map<ffi::String, runtime::Tensor>>
params;
48 refl::ObjectDef<BuilderInputNode>()
71 ffi::Optional<ffi::Map<ffi::String, runtime::Tensor>> params = std::nullopt);
85 refl::ObjectDef<BuilderResultNode>()
105 ffi::Optional<ffi::String> error_msg);
120 virtual ffi::Array<BuilderResult>
Build(
const ffi::Array<BuilderInput>& build_inputs) = 0;
126 using FBuild = ffi::TypedFunction<ffi::Array<BuilderResult>(
const ffi::Array<BuilderInput>&)>;
142 explicit Builder(ObjectPtr<BuilderNode> data) : ObjectRef(data) {
143 TVM_FFI_ICHECK(data !=
nullptr);
165 ffi::Array<BuilderResult>
Build(
const ffi::Array<BuilderInput>& build_inputs)
final {
166 ICHECK(
f_build !=
nullptr) <<
"PyBuilder's Build method not implemented!";
Managed reference class to IRModuleNode.
Definition: module.h:256
Managed reference class to TargetNode.
Definition: target.h:192
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.
A device-independent managed Tensor abstraction.
Compilation target object.