diff --git a/include/tvm/runtime/crt/module.h b/include/tvm/runtime/crt/module.h index b825f6d55ed8..2359025f6fe1 100644 --- a/include/tvm/runtime/crt/module.h +++ b/include/tvm/runtime/crt/module.h @@ -27,6 +27,10 @@ #include #include +#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 index 4a927340467b..a0558268dbfb 100644 --- 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('0' + ((c >> 6) & 0x03)) + << static_cast('0' + ((c >> 3) & 0x07)) + << static_cast('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 index 3f1e3aaef933..f4aa7281f1f9 100644 --- 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 \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"; }