Skip to content

Commit

Permalink
Fix tvm_module_wrapper build error
Browse files Browse the repository at this point in the history
[100%] Building CXX object CMakeFiles/pt_tvmdsoop_new.dir/src/contrib/torch/tvm_module_wrapper/RuntimeModuleWrapperTVM.cc.o
mlc-llm/3rdparty/tvm/src/contrib/torch/tvm_module_wrapper/RuntimeModuleWrapperTVM.cc:32:10: fatal error: ../../support/base64.h: No such file or directory
   32 | #include "../../support/base64.h"
      |          ^~~~~~~~~~~~~~~~~~~~~~~~

Signed-off-by: Lei Wen <wenlei03@qiyi.com>
  • Loading branch information
wenlei03 committed Sep 12, 2023
1 parent cf5b492 commit c3573c5
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions src/contrib/torch/tvm_module_wrapper/RuntimeModuleWrapperTVM.cc
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
#include <vector>

#include "../../../runtime/graph_executor/graph_executor_factory.h"
#include "../../support/base64.h"
#include "../../../support/base64.h"
#include "runtime_bridge.h"

namespace tvm {
Expand Down Expand Up @@ -209,10 +209,10 @@ inline void b64decode(const std::string b64str, uint8_t* ret) {
size_t index = 0;
const auto length = b64str.size();
for (size_t i = 0; i < length; i += 4) {
int8_t ch0 = base64::DecodeTable[(int32_t)b64str[i]];
int8_t ch1 = base64::DecodeTable[(int32_t)b64str[i + 1]];
int8_t ch2 = base64::DecodeTable[(int32_t)b64str[i + 2]];
int8_t ch3 = base64::DecodeTable[(int32_t)b64str[i + 3]];
int8_t ch0 = tvm::support::base64::DecodeTable[(int32_t)b64str[i]];
int8_t ch1 = tvm::support::base64::DecodeTable[(int32_t)b64str[i + 1]];
int8_t ch2 = tvm::support::base64::DecodeTable[(int32_t)b64str[i + 2]];
int8_t ch3 = tvm::support::base64::DecodeTable[(int32_t)b64str[i + 3]];
uint8_t st1 = (ch0 << 2) + (ch1 >> 4);
ret[index++] = st1;
if (b64str[i + 2] != '=') {
Expand Down

0 comments on commit c3573c5

Please sign in to comment.