Rename tvm_dso_op to libtvm_dso_op (#5714)
authortobe <tobeg3oogle@gmail.com>
Tue, 2 Jun 2020 16:22:42 +0000 (00:22 +0800)
committerGitHub <noreply@github.com>
Tue, 2 Jun 2020 16:22:42 +0000 (09:22 -0700)
cmake/modules/contrib/TF_TVMDSOOP.cmake
python/tvm/contrib/tf_op/module.py

index e92822a..1509e83 100644 (file)
@@ -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")
index fd0ed0c..7daf45f 100644 (file)
@@ -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):