Skip to content

[BREAKING CHANGE] Prefix all type keys and function names with "tvm." #18611

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
junrushao wants to merge 1 commit into
base: main
Choose a base branch
from
Open

[BREAKING CHANGE] Prefix all type keys and function names with "tvm." #18611

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
2 changes: 1 addition & 1 deletion apps/cpp_rpc/rpc_server.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -401,7 +401,7 @@ void RPCServerCreate(std::string host, int port, int port_end, std::string track

TVM_FFI_STATIC_INIT_BLOCK() {
namespace refl = tvm::ffi::reflection;
refl::GlobalDef().def("rpc.ServerCreate", RPCServerCreate);
refl::GlobalDef().def("tvm.rpc.ServerCreate", RPCServerCreate);
}
} // namespace runtime
} // namespace tvm
7 changes: 4 additions & 3 deletions apps/hexagon_launcher/launcher_core.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -148,12 +148,13 @@ const tvm::ffi::Function get_module_func(tvm::runtime::Module module, const std:
}

void reset_device_api() {
const tvm::ffi::Function api = get_runtime_func("device_api.hexagon");
tvm::ffi::Function::SetGlobal("device_api.cpu", api, true);
const tvm::ffi::Function api = get_runtime_func("tvm.device_api.hexagon");
tvm::ffi::Function::SetGlobal("tvm.device_api.cpu", api, true);
}

tvm::runtime::Module load_module(const std::string& file_name) {
static const tvm::ffi::Function loader = get_runtime_func("ffi.Module.load_from_file.hexagon");
static const tvm::ffi::Function loader =
get_runtime_func("tvm.ffi.Module.load_from_file.hexagon");
tvm::ffi::Any rv = loader(file_name);
if (rv.type_code() == kTVMModuleHandle) {
ICHECK_EQ(rv.type_code(), kTVMModuleHandle)
Expand Down
4 changes: 2 additions & 2 deletions apps/hexagon_launcher/launcher_hexagon.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,15 +48,15 @@ int __QAIC_HEADER(launcher_rpc_open)(const char* uri, remote_handle64* handle) {
*handle = 0; // Just use any value.
reset_device_api();
static const tvm::ffi::Function acq_res =
get_runtime_func("device_api.hexagon.acquire_resources");
get_runtime_func("tvm.device_api.hexagon.acquire_resources");
acq_res();
return AEE_SUCCESS;
}

int __QAIC_HEADER(launcher_rpc_close)(remote_handle64 handle) {
// Comment to stop clang-format from single-lining this function.
static const tvm::ffi::Function rel_res =
get_runtime_func("device_api.hexagon.release_resources");
get_runtime_func("tvm.device_api.hexagon.release_resources");
rel_res();
return AEE_SUCCESS;
}
Expand Down
2 changes: 1 addition & 1 deletion apps/ios_rpc/tests/ios_rpc_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@


# override metal compiler to compile to iphone
@tvm.register_global_func("tvm_callback_metal_compile")
@tvm.register_global_func("tvm.tvm_callback_metal_compile")
Copy link
Contributor

@gemini-code-assist gemini-code-assist bot Dec 25, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

The new function name tvm.tvm_callback_metal_compile seems redundant. To improve clarity and consistency with the new naming scheme, consider renaming it to tvm.callback_metal_compile. This would make the naming scheme more consistent by avoiding the repeated tvm prefix.

Suggested change
@tvm.register_global_func("tvm.tvm_callback_metal_compile")
@tvm.register_global_func("tvm.callback_metal_compile")

def compile_metal(src, target):
return xcode.compile_metal(src, sdk=sdk)

Expand Down
2 changes: 1 addition & 1 deletion apps/ios_rpc/tvmrpc/RPCServer.mm
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@
*/
FEventHandler CreateServerEventHandler(NSOutputStream* outputStream, std::string name,
std::string remote_key) {
auto event_handler_factory = tvm::ffi::Function::GetGlobal("rpc.CreateEventDrivenServer");
auto event_handler_factory = tvm::ffi::Function::GetGlobal("tvm.rpc.CreateEventDrivenServer");
ICHECK(event_handler_factory.has_value())
<< "You are using tvm_runtime module built without RPC support. "
<< "Please rebuild it with USE_RPC flag.";
Expand Down
4 changes: 2 additions & 2 deletions apps/ios_rpc/tvmrpc/TVMRuntime.mm
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ void LogMessageImpl(const std::string& file, int lineno, int level, const std::s
NSBundle* bundle = [NSBundle mainBundle];
base = [[bundle privateFrameworksPath] stringByAppendingPathComponent:@"tvm"];

if (tvm::ffi::Function::GetGlobal("ffi.Module.load_from_file.dylib_custom")) {
if (tvm::ffi::Function::GetGlobal("tvm.ffi.Module.load_from_file.dylib_custom")) {
// Custom dso loader is present. Will use it.
base = NSTemporaryDirectory();
fmt = "dylib_custom";
Expand Down Expand Up @@ -114,7 +114,7 @@ void Init(const std::string& name) {
// Add UnsignedDSOLoader plugin in global registry
TVM_FFI_STATIC_INIT_BLOCK() {
namespace refl = tvm::ffi::reflection;
refl::GlobalDef().def_packed("ffi.Module.load_from_file.dylib_custom",
refl::GlobalDef().def_packed("tvm.ffi.Module.load_from_file.dylib_custom",
[](ffi::PackedArgs args, ffi::Any* rv) {
auto n = ffi::make_object<UnsignedDSOLoader>();
n->Init(args[0]);
Expand Down
4 changes: 2 additions & 2 deletions include/tvm/arith/analyzer.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,7 @@ class ConstIntBoundNode : public Object {
static const constexpr int64_t kNegInf = -kPosInf;

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.ConstIntBound", ConstIntBoundNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.ConstIntBound", ConstIntBoundNode, Object);
};

/*!
Expand Down Expand Up @@ -215,7 +215,7 @@ class ModularSetNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.ModularSet", ModularSetNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.ModularSet", ModularSetNode, Object);
};

/*!
Expand Down
2 changes: 1 addition & 1 deletion include/tvm/arith/int_set.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ enum SignType { kPositive, kNegative, kZero, kUnknown };
*/
class IntSetNode : public Object {
public:
TVM_FFI_DECLARE_OBJECT_INFO("ir.IntSet", IntSetNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.IntSet", IntSetNode, Object);
};

/*!
Expand Down
8 changes: 4 additions & 4 deletions include/tvm/arith/int_solver.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ class IntGroupBoundsNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IntGroupBounds", IntGroupBoundsNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IntGroupBounds", IntGroupBoundsNode, Object);
};

/*!
Expand Down Expand Up @@ -150,7 +150,7 @@ class IntConstraintsNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IntConstraints", IntConstraintsNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IntConstraints", IntConstraintsNode, Object);
};

/*!
Expand Down Expand Up @@ -203,8 +203,8 @@ class IntConstraintsTransformNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IntConstraintsTransform", IntConstraintsTransformNode,
Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IntConstraintsTransform",
IntConstraintsTransformNode, Object);
};

/*!
Expand Down
10 changes: 5 additions & 5 deletions include/tvm/arith/iter_affine_map.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ namespace arith {
class IterMapExprNode : public PrimExprNode {
public:
static constexpr const uint32_t _type_child_slots = 2;
TVM_FFI_DECLARE_OBJECT_INFO("arith.IterMapExpr", IterMapExprNode, PrimExprNode);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.arith.IterMapExpr", IterMapExprNode, PrimExprNode);
};

/*!
Expand Down Expand Up @@ -105,7 +105,7 @@ class IterMarkNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindDAGNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IterMark", IterMarkNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IterMark", IterMarkNode, Object);
};

/*!
Expand Down Expand Up @@ -151,7 +151,7 @@ class IterSplitExprNode : public IterMapExprNode {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IterSplitExpr", IterSplitExprNode, IterMapExprNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IterSplitExpr", IterSplitExprNode, IterMapExprNode);
};

/*!
Expand Down Expand Up @@ -205,7 +205,7 @@ class IterSumExprNode : public IterMapExprNode {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IterSumExpr", IterSumExprNode, IterMapExprNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IterSumExpr", IterSumExprNode, IterMapExprNode);
};

/*!
Expand Down Expand Up @@ -264,7 +264,7 @@ class IterMapResultNode : public Object {
.def_ro("errors", &IterMapResultNode::errors)
.def_ro("padding_predicate", &IterMapResultNode::padding_predicate);
}
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("arith.IterMapResult", IterMapResultNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.arith.IterMapResult", IterMapResultNode, Object);
};

/*!
Expand Down
6 changes: 3 additions & 3 deletions include/tvm/ir/attrs.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ class AttrFieldInfoNode : public Object {

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;

TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.AttrFieldInfo", AttrFieldInfoNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.AttrFieldInfo", AttrFieldInfoNode, Object);
};

/*! \brief AttrFieldInfo */
Expand Down Expand Up @@ -121,7 +121,7 @@ class BaseAttrsNode : public Object {
bool allow_unknown = false) = 0;

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO("ir.Attrs", BaseAttrsNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.Attrs", BaseAttrsNode, Object);
};

/*!
Expand Down Expand Up @@ -152,7 +152,7 @@ class DictAttrsNode : public BaseAttrsNode {
void InitByPackedArgs(const ffi::PackedArgs& args, bool allow_unknown) final;

// type info
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.DictAttrs", DictAttrsNode, BaseAttrsNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.DictAttrs", DictAttrsNode, BaseAttrsNode);
};

/*!
Expand Down
6 changes: 3 additions & 3 deletions include/tvm/ir/diagnostic.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ class DiagnosticNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("Diagnostic", DiagnosticNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.Diagnostic", DiagnosticNode, Object);
};

class Diagnostic : public ObjectRef {
Expand Down Expand Up @@ -166,7 +166,7 @@ class DiagnosticRendererNode : public Object {
namespace refl = tvm::ffi::reflection;
refl::ObjectDef<DiagnosticRendererNode>().def_ro("renderer", &DiagnosticRendererNode::renderer);
}
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("DiagnosticRenderer", DiagnosticRendererNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.DiagnosticRenderer", DiagnosticRendererNode, Object);
};

class DiagnosticRenderer : public ObjectRef {
Expand Down Expand Up @@ -205,7 +205,7 @@ class DiagnosticContextNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("DiagnosticContext", DiagnosticContextNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.DiagnosticContext", DiagnosticContextNode, Object);
};

class DiagnosticContext : public ObjectRef {
Expand Down
2 changes: 1 addition & 1 deletion include/tvm/ir/env_func.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ class EnvFuncNode : public Object {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.EnvFunc", EnvFuncNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.EnvFunc", EnvFuncNode, Object);
};

/*!
Expand Down
16 changes: 8 additions & 8 deletions include/tvm/ir/expr.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ class BaseExprNode : public Object {
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;

static constexpr const uint32_t _type_child_slots = 64;
TVM_FFI_DECLARE_OBJECT_INFO("ir.BaseExpr", BaseExprNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.BaseExpr", BaseExprNode, Object);
};

/*!
Expand Down Expand Up @@ -114,7 +114,7 @@ class PrimExprNode : public BaseExprNode {
TVM_OBJECT_ENABLE_SCRIPT_PRINTER();

static constexpr const uint32_t _type_child_slots = 40;
TVM_FFI_DECLARE_OBJECT_INFO("ir.PrimExpr", PrimExprNode, BaseExprNode);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.PrimExpr", PrimExprNode, BaseExprNode);
};

/*!
Expand Down Expand Up @@ -155,7 +155,7 @@ class PrimExprConvertibleNode : public Object {
public:
virtual ~PrimExprConvertibleNode() {}
virtual PrimExpr ToPrimExpr() const = 0;
TVM_FFI_DECLARE_OBJECT_INFO("ir.PrimExprConvertible", PrimExprConvertibleNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.PrimExprConvertible", PrimExprConvertibleNode, Object);
};

/*!
Expand Down Expand Up @@ -429,7 +429,7 @@ class RelaxExprNode : public BaseExprNode {
}

static constexpr const uint32_t _type_child_slots = 22;
TVM_FFI_DECLARE_OBJECT_INFO("ir.RelaxExpr", RelaxExprNode, BaseExprNode);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.RelaxExpr", RelaxExprNode, BaseExprNode);
};

/*!
Expand Down Expand Up @@ -471,7 +471,7 @@ class GlobalVarNode : public RelaxExprNode {
}

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindFreeVar;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.GlobalVar", GlobalVarNode, RelaxExprNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.GlobalVar", GlobalVarNode, RelaxExprNode);
};

/*!
Expand Down Expand Up @@ -499,7 +499,7 @@ class IntImmNode : public PrimExprNode {
namespace refl = tvm::ffi::reflection;
refl::ObjectDef<IntImmNode>().def_ro("value", &IntImmNode::value);
}
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.IntImm", IntImmNode, PrimExprNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.IntImm", IntImmNode, PrimExprNode);
};

/*!
Expand Down Expand Up @@ -534,7 +534,7 @@ class FloatImmNode : public PrimExprNode {
namespace refl = tvm::ffi::reflection;
refl::ObjectDef<FloatImmNode>().def_ro("value", &FloatImmNode::value);
}
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.FloatImm", FloatImmNode, PrimExprNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.FloatImm", FloatImmNode, PrimExprNode);
};

/*!
Expand Down Expand Up @@ -682,7 +682,7 @@ class RangeNode : public Object {

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;

TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.Range", RangeNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.Range", RangeNode, Object);
};

/*! \brief Range container */
Expand Down
2 changes: 1 addition & 1 deletion include/tvm/ir/function.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -223,7 +223,7 @@ class BaseFuncNode : public RelaxExprNode {
}

static constexpr const uint32_t _type_child_slots = 2;
TVM_FFI_DECLARE_OBJECT_INFO("ir.BaseFunc", BaseFuncNode, RelaxExprNode);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.BaseFunc", BaseFuncNode, RelaxExprNode);
};

/*!
Expand Down
6 changes: 3 additions & 3 deletions include/tvm/ir/global_info.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ class GlobalInfoNode : public Object {
public:
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;

TVM_FFI_DECLARE_OBJECT_INFO("ir.GlobalInfo", GlobalInfoNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.ir.GlobalInfo", GlobalInfoNode, Object);
};

/*!
Expand Down Expand Up @@ -77,7 +77,7 @@ class VDeviceNode : public GlobalInfoNode {
.def_ro("memory_scope", &VDeviceNode::memory_scope);
}

TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.VDevice", VDeviceNode, GlobalInfoNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.VDevice", VDeviceNode, GlobalInfoNode);
};

/*!
Expand All @@ -100,7 +100,7 @@ class DummyGlobalInfoNode : public GlobalInfoNode {
refl::ObjectDef<DummyGlobalInfoNode>();
}

TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.DummyGlobalInfo", DummyGlobalInfoNode, GlobalInfoNode);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.DummyGlobalInfo", DummyGlobalInfoNode, GlobalInfoNode);
};

/*!
Expand Down
2 changes: 1 addition & 1 deletion include/tvm/ir/global_var_supply.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ class GlobalVarSupplyNode : public Object {
NameSupply name_supply_;

static constexpr const bool _type_mutable = true;
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.GlobalVarSupply", GlobalVarSupplyNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.GlobalVarSupply", GlobalVarSupplyNode, Object);

private:
std::unordered_map<std::string, GlobalVar> name_to_var_map_;
Expand Down
2 changes: 1 addition & 1 deletion include/tvm/ir/instrument.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ class PassInstrumentNode : public Object {
namespace refl = tvm::ffi::reflection;
refl::ObjectDef<PassInstrumentNode>().def_ro("name", &PassInstrumentNode::name);
}
TVM_FFI_DECLARE_OBJECT_INFO("instrument.PassInstrument", PassInstrumentNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO("tvm.instrument.PassInstrument", PassInstrumentNode, Object);
};

/*!
Expand Down
6 changes: 3 additions & 3 deletions include/tvm/ir/module.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,8 +140,8 @@ class IRModuleNode : public Object {
.def_ro("global_infos", &IRModuleNode::global_infos);
// register custom structural equal and hash.
refl::TypeAttrDef<IRModuleNode>()
.def("__s_equal__", &IRModuleNode::SEqual)
.def("__s_hash__", &IRModuleNode::SHash);
.def("tvm.__s_equal__", &IRModuleNode::SEqual)
.def("tvm.__s_hash__", &IRModuleNode::SHash);
}

TVM_DLL bool SEqual(const IRModuleNode* other,
Expand Down Expand Up @@ -243,7 +243,7 @@ class IRModuleNode : public Object {

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;

TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.IRModule", IRModuleNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.IRModule", IRModuleNode, Object);

private:
friend class IRModule;
Expand Down
2 changes: 1 addition & 1 deletion include/tvm/ir/name_supply.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ class NameSupplyNode : public Object {
refl::ObjectDef<NameSupplyNode>();
}

TVM_FFI_DECLARE_OBJECT_INFO_FINAL("ir.NameSupply", NameSupplyNode, Object);
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tvm.ir.NameSupply", NameSupplyNode, Object);

private:
/*! \brief Helper function to add the NameSupply prefix to the name. */
Expand Down
Loading
Loading