2022-02-15 09:21:02 +00:00
|
|
|
if (onnxruntime_USE_TVM)
|
|
|
|
|
message(STATUS "onnxruntime_USE_TVM: Fetch tvm for TVM EP")
|
2022-01-27 19:31:13 +00:00
|
|
|
|
|
|
|
|
FetchContent_Declare(
|
|
|
|
|
tvm
|
|
|
|
|
GIT_REPOSITORY https://github.com/apache/tvm.git
|
2022-07-13 08:48:42 +00:00
|
|
|
GIT_TAG 3425ed846308a456f98404c79f6df1693bed6377
|
2022-01-27 19:31:13 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
FetchContent_GetProperties(tvm)
|
|
|
|
|
if(NOT tvm_POPULATED)
|
|
|
|
|
FetchContent_Populate(tvm)
|
2022-07-13 08:48:42 +00:00
|
|
|
if (WIN32)
|
|
|
|
|
execute_process(
|
|
|
|
|
COMMAND ${CMAKE_COMMAND} -E create_symlink ${tvm_BINARY_DIR}/${CMAKE_BUILD_TYPE} ${tvm_SOURCE_DIR}/build
|
|
|
|
|
)
|
|
|
|
|
else()
|
|
|
|
|
file(CREATE_LINK ${tvm_BINARY_DIR} ${tvm_SOURCE_DIR}/build SYMBOLIC)
|
|
|
|
|
endif()
|
2022-01-27 19:31:13 +00:00
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
set(tvm_INCLUDE_DIRS ${tvm_SOURCE_DIR}/include)
|
|
|
|
|
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
if (onnxruntime_USE_NUPHAR)
|
2022-02-15 09:21:02 +00:00
|
|
|
message(STATUS "onnxruntime_USE_NUPHAR: Fetch onnxruntime-tvm for NUPHAR EP")
|
2022-01-27 19:31:13 +00:00
|
|
|
|
|
|
|
|
FetchContent_Declare(
|
|
|
|
|
tvm
|
|
|
|
|
GIT_REPOSITORY https://github.com/microsoft/onnxruntime-tvm.git
|
|
|
|
|
GIT_TAG 9ec2b92d180dff8877e402018b97baa574031b8b
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
FetchContent_GetProperties(tvm)
|
|
|
|
|
if(NOT tvm_POPULATED)
|
|
|
|
|
FetchContent_Populate(tvm)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
set(tvm_INCLUDE_DIRS ${tvm_SOURCE_DIR}/include)
|
|
|
|
|
|
2022-06-04 07:31:01 +00:00
|
|
|
endif()
|