Rename tvm_dso_op to libtvm_dso_op (apache#5714)

trevor-m · Jun 9, 2020 · c29f3b7 · c29f3b7
1 parent c8ab3f6
commit c29f3b7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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):