mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
7cb4b9b18a
* finish framework * add autotvm and auto-scheduler tuner * add python deploy api * add SDK net(WIP * add sdk support * support det, support vm * fix vm sdk * support two stage detector * add instance seg support * add docstring * update docs and ut * add quantize * update doc * update docs * synchronize stream * support dlpack * remove submodule * fix stride * add alignment * support dlpack * remove submodule * replace exclusive_scan * add backend check * add build script * fix comment * add ci * fix ci * ci fix2 * update build script * update ci * add pytest * update sed command * update sed again * add xgboost * remove tvm ut * update ansor runner * add stream sync * fix topk * sync default stream * fix tvm net * fix window
48 lines
1.4 KiB
CMake
48 lines
1.4 KiB
CMake
# Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
if (NOT DEFINED TVM_DIR)
|
|
set(TVM_DIR $ENV{TVM_DIR})
|
|
endif ()
|
|
if (NOT TVM_DIR)
|
|
message(FATAL_ERROR "Please set TVM_DIR with cmake -D option.")
|
|
endif()
|
|
|
|
find_path(
|
|
TVM_INCLUDE_DIR tvm/runtime/c_runtime_api.h
|
|
HINTS ${TVM_DIR}
|
|
PATH_SUFFIXES include)
|
|
|
|
find_path(
|
|
DMLC_CORE_INCLUDE_DIR dmlc/io.h
|
|
HINTS ${TVM_DIR}/3rdparty/dmlc-core
|
|
PATH_SUFFIXES include)
|
|
|
|
find_path(
|
|
DLPACK_INCLUDE_DIR dlpack/dlpack.h
|
|
HINTS ${TVM_DIR}/3rdparty/dlpack
|
|
PATH_SUFFIXES include)
|
|
|
|
find_library(
|
|
TVM_LIBRARY_PATH tvm_runtime
|
|
HINTS ${TVM_DIR}
|
|
PATH_SUFFIXES build lib build/${CMAKE_BUILD_TYPE})
|
|
if (NOT (TVM_INCLUDE_DIR AND DMLC_CORE_INCLUDE_DIR AND DLPACK_INCLUDE_DIR AND TVM_LIBRARY_PATH))
|
|
message(FATAL_ERROR "Couldn't find tvm in TVM_DIR: "
|
|
"${TVM_DIR}, please check if the path is correct.")
|
|
endif()
|
|
|
|
add_library(tvm_runtime SHARED IMPORTED)
|
|
set_property(TARGET tvm_runtime APPEND PROPERTY IMPORTED_CONFIGURATIONS RELEASE)
|
|
if (MSVC)
|
|
set_target_properties(tvm_runtime PROPERTIES
|
|
IMPORTED_IMPLIB_RELEASE ${TVM_LIBRARY_PATH}
|
|
INTERFACE_INCLUDE_DIRECTORIES ${TVM_INCLUDE_DIR} ${DMLC_CORE_INCLUDE_DIR} ${DLPACK_INCLUDE_DIR}
|
|
)
|
|
|
|
else()
|
|
set_target_properties(tvm_runtime PROPERTIES
|
|
IMPORTED_LOCATION_RELEASE ${TVM_LIBRARY_PATH}
|
|
INTERFACE_INCLUDE_DIRECTORIES ${TVM_INCLUDE_DIR} ${DMLC_CORE_INCLUDE_DIR} ${DLPACK_INCLUDE_DIR}
|
|
)
|
|
endif()
|