You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2022/06/09 11:42:02 UTC
[GitHub] [tvm] zhuwenxi commented on a diff in pull request #11326: [LIBXSMM][BYOC] Integrate Libxsmm with BYOC
zhuwenxi commented on code in PR #11326:
URL: https://github.com/apache/tvm/pull/11326#discussion_r893398981
##########
cmake/modules/contrib/LIBXSMM.cmake:
##########
@@ -0,0 +1,31 @@
+if(IS_DIRECTORY ${USE_LIBXSMM})
+ find_library(LIBXSMM_LIBRARY NAMES xsmm HINTS ${USE_LIBXSMM}/lib/)
+ if(LIBXSMM_LIBRARY STREQUAL "LIBXSMM_LIBRARY-NOTFOUND")
+ message(WARNING "Cannot find LIBXSMM library at ${USE_LIBXSMM}/lib/.")
+ else()
+ include_directories(SYSTEM ${USE_LIBXSMM}/include)
+ list(APPEND TVM_RUNTIME_LINKER_LIBS ${LIBXSMM_LIBRARY})
+ MESSAGE(STATUS "Use LIBXSMM library " ${LIBXSMM_LIBRARY})
+
+ tvm_file_glob(GLOB LIBXSMM_RELAY_CONTRIB_SRC src/relay/backend/contrib/libxsmm/*.cc)
+ list(APPEND COMPILER_SRCS ${LIBXSMM_RELAY_CONTRIB_SRC})
+ tvm_file_glob(GLOB LIBXSMM_RUNTIME_SRC src/runtime/contrib/libxsmm/*.cc)
+ list(APPEND RUNTIME_SRCS ${LIBXSMM_RUNTIME_SRC})
+ endif()
+elseif(USE_LIBXSMM STREQUAL "ON")
+ find_library(LIBXSMM_LIBRARY xsmm)
+ if(LIBXSMM_LIBRARY STREQUAL "LIBXSMM_LIBRARY-NOTFOUND")
+ message(WARNING "Cannot find LIBXSMM library at $(USE_LIBXSMM).")
+ else()
+ list(APPEND TVM_RUNTIME_LINKER_LIBS ${LIBXSMM_LIBRARY})
+ MESSAGE(STATUS "Use LIBXSMM library " ${LIBXSMM_LIBRARY})
+
+ tvm_file_glob(GLOB LIBXSMM_RELAY_CONTRIB_SRC src/relay/backend/contrib/libxsmm/*.cc)
+ list(APPEND COMPILER_SRCS ${LIBXSMM_RELAY_CONTRIB_SRC})
+ tvm_file_glob(GLOB LIBXSMM_RUNTIME_SRC src/runtime/contrib/libxsmm/*.cc)
+ list(APPEND RUNTIME_SRCS ${LIBXSMM_RUNTIME_SRC})
+ endif()
Review Comment:
I'm not sure which part could be merged. I followed the flavor of https://github.com/apache/tvm/blob/main/cmake/modules/contrib/DNNL.cmake exactly. From my point of view, only those 4 `tvm_file_glob` lines could be merged. Do you have any sugguestions?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org