[tvm] branch main updated: Use a uint64_t to serialize primitive_attrs in the Relay VM to fix 32bit RPC (#9169)

[mbrookhart]

This is an automated email from the ASF dual-hosted git repository.

mbrookhart pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 61fbda9  Use a uint64_t to serialize primitive_attrs in the Relay VM 
to fix 32bit RPC (#9169)
61fbda9 is described below

commit 61fbda9394dbe88f85bcb2789efd5b2f4cb191ee
Author: Matthew Brookhart <mbrookh...@octoml.ai>
AuthorDate: Fri Oct 1 08:07:44 2021 -0600

    Use a uint64_t to serialize primitive_attrs in the Relay VM to fix 32bit 
RPC (#9169)
---
 src/runtime/vm/executable.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/runtime/vm/executable.cc b/src/runtime/vm/executable.cc
index c2dc030..a5e7d25 100644
--- a/src/runtime/vm/executable.cc
+++ b/src/runtime/vm/executable.cc
@@ -273,7 +273,7 @@ void Executable::SavePrimitiveOpNames(dmlc::Stream* strm) {
     primitive_names[packed_index] = it.first;
   }
   strm->Write(primitive_names);
-  std::map<size_t, std::map<std::string, std::string>> primitive_attrs;
+  std::map<uint64_t, std::map<std::string, std::string>> primitive_attrs;
   for (const auto& it : this->op_attrs) {
     auto packed_index = static_cast<size_t>(it.first);
     std::map<std::string, std::string> attrs;
@@ -584,7 +584,7 @@ void Executable::LoadPrimitiveOpNames(dmlc::Stream* strm) {
     this->primitive_map.insert({primitive_names[i], i});
   }
 
-  std::map<size_t, std::map<std::string, std::string>> primitive_attrs;
+  std::map<uint64_t, std::map<std::string, std::string>> primitive_attrs;
   STREAM_CHECK(strm->Read(&primitive_attrs), "primitive attrs");
   for (const auto& fn : primitive_attrs) {
     std::vector<std::pair<String, ObjectRef>> attrs;