[uTVM] fix crt building and running error (apache#6231)

Signed-off-by: windclarion <[email protected]>
neo-ai · Sep 3, 2020 · 09a3f16 · 09a3f16
1 parent a82d46d
commit 09a3f16
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/include/tvm/runtime/crt/module.h b/include/tvm/runtime/crt/module.h
@@ -27,6 +27,10 @@
 #include <tvm/runtime/c_backend_api.h>
 #include <tvm/runtime/crt/func_registry.h>
 
+#ifdef __cplusplus
+extern "C" {
+#endif
+
 /*!
  * \brief Module container of TVM.
  */
@@ -38,4 +42,7 @@ typedef struct TVMModule {
 /*! \brief Entry point for the system lib module. */
 const TVMModule* TVMSystemLibEntryPoint(void);
 
+#ifdef __cplusplus
+}
+#endif
 #endif  // TVM_RUNTIME_CRT_MODULE_H_
diff --git a/src/support/str_escape.h b/src/support/str_escape.h
@@ -65,7 +65,9 @@ inline std::string StrEscape(const char* data, size_t size, bool use_octal_escap
           break;
         default:
           if (use_octal_escape) {
-            stream << '0' + ((c >> 6) & 0x03) << '0' + ((c >> 3) & 0x07) << '0' + (c & 0x03);
+            stream << static_cast<unsigned char>('0' + ((c >> 6) & 0x03))
+                   << static_cast<unsigned char>('0' + ((c >> 3) & 0x07))
+                   << static_cast<unsigned char>('0' + (c & 0x07));
           } else {
             const char* hex_digits = "0123456789ABCDEF";
             stream << 'x' << hex_digits[c >> 4] << hex_digits[c & 0xf];

diff --git a/src/target/source/codegen_c_host.cc b/src/target/source/codegen_c_host.cc
@@ -279,8 +279,9 @@ void CodeGenCHost::GenerateFuncRegistry() {
   decl_stream << "#include <tvm/runtime/crt/module.h>\n";
   stream << "static TVMBackendPackedCFunc _tvm_func_array[] = {\n";
   for (auto f : function_names_) {
-    stream << "    " << f << ",\n";
+    stream << "    (TVMBackendPackedCFunc)" << f << ",\n";
   }
+  stream << "};\n";
   auto registry = target::GenerateFuncRegistryNames(function_names_);
   stream << "static const TVMFuncRegistry _tvm_func_registry = {\n"
          << "    \"" << ::tvm::support::StrEscape(registry.data(), registry.size(), true) << "\","
@@ -290,10 +291,10 @@ void CodeGenCHost::GenerateFuncRegistry() {
 
 void CodeGenCHost::GenerateCrtSystemLib() {
   stream << "static const TVMModule _tvm_system_lib = {\n"
-         << "    &system_lib_registry,\n"
+         << "    &_tvm_func_registry,\n"
          << "};\n"
          << "const TVMModule* TVMSystemLibEntryPoint(void) {\n"
-         << "    return &system_lib;\n"
+         << "    return &_tvm_system_lib;\n"
          << "}\n";
 }