From 1832e049e8a742c644ace8e7fa1993a2d6dc535e Mon Sep 17 00:00:00 2001 From: neucer Date: Thu, 14 Sep 2023 15:22:25 +0300 Subject: [PATCH] [BugFix] Move symbols that are relevant to the runtime from libtvm to libtvm-runtime --- cmake/modules/contrib/LibTorch.cmake | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/cmake/modules/contrib/LibTorch.cmake b/cmake/modules/contrib/LibTorch.cmake index 0e00e7008c2f..85881410ade4 100644 --- a/cmake/modules/contrib/LibTorch.cmake +++ b/cmake/modules/contrib/LibTorch.cmake @@ -23,8 +23,12 @@ if(USE_LIBTORCH) file(GLOB LIBTORCH_RELAY_CONTRIB_SRC src/relay/backend/contrib/libtorch/libtorch_codegen.cc - src/runtime/contrib/libtorch/libtorch_runtime.cc ) list(APPEND COMPILER_SRCS ${LIBTORCH_RELAY_CONTRIB_SRC}) + file(GLOB LIBTORCH_RUNTIME_CONTRIB_SRC + src/runtime/contrib/libtorch/libtorch_runtime.cc + ) + list(APPEND RUNTIME_SRCS ${LIBTORCH_RUNTIME_CONTRIB_SRC}) + endif(USE_LIBTORCH)