diff --git a/cmake/modules/contrib/TF_TVMDSOOP.cmake b/cmake/modules/contrib/TF_TVMDSOOP.cmake index e92822a397ae..1509e83a9be3 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 fd0ed0c8fb2f..7daf45fcb0c2 100644 --- a/python/tvm/contrib/tf_op/module.py +++ b/python/tvm/contrib/tf_op/module.py @@ -68,7 +68,7 @@ def __init__(self, lib_path, func_name, output_dtype, output_shape): 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):