From 062a244d66262353cdef0792a54d05cc99d7fa74 Mon Sep 17 00:00:00 2001 From: tobe Date: Wed, 3 Jun 2020 00:22:42 +0800 Subject: [PATCH] Rename tvm_dso_op to libtvm_dso_op (#5714) --- cmake/modules/contrib/TF_TVMDSOOP.cmake | 1 - python/tvm/contrib/tf_op/module.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/cmake/modules/contrib/TF_TVMDSOOP.cmake b/cmake/modules/contrib/TF_TVMDSOOP.cmake index e92822a..1509e83 100644 --- a/cmake/modules/contrib/TF_TVMDSOOP.cmake +++ b/cmake/modules/contrib/TF_TVMDSOOP.cmake @@ -44,7 +44,6 @@ if(NOT USE_TF_TVMDSOOP STREQUAL "OFF") set(OP_LIBRARY_NAME tvm_dso_op) file(GLOB_RECURSE TFTVM_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/src/contrib/tf_op/*.cc) add_library(${OP_LIBRARY_NAME} SHARED ${TFTVM_SRCS}) - set_target_properties(${OP_LIBRARY_NAME} PROPERTIES PREFIX "") set(TFTVM_LINK_FLAGS -ltvm -L${CMAKE_CURRENT_BINARY_DIR}) if (NOT BUILD_TVMDSOOP_ONLY STREQUAL "ON") diff --git a/python/tvm/contrib/tf_op/module.py b/python/tvm/contrib/tf_op/module.py index fd0ed0c..7daf45f 100644 --- a/python/tvm/contrib/tf_op/module.py +++ b/python/tvm/contrib/tf_op/module.py @@ -68,7 +68,7 @@ class TensorFunc: elif output_shape is not None: self.dynamic_output_shape = self._pack_shape_tensor(output_shape) - self.module = self._load_platform_specific_library("tvm_dso_op") + self.module = self._load_platform_specific_library("libtvm_dso_op") self.tvm_dso_op = self.module.tvm_dso_op def apply(self, *params): -- 2.7.4