mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-31 23:27:43 +00:00
62cc981599
* add executor option (vm or graph) and support virtual machine methods * nullptr check for compile and run methods (see also PR#10211 from microsoft:onnxruntime) * get output shapes for VM * remove run_with_benchmark. remove run methods from python api, get it from native side * get outputs method for VM was implemented * support multiple input for VM * update python logging and exception * small fix * update tvm with patch for VM API * update nhwc transformations for TVM EP * add data alignment check and support set_input_zero_copy for GE in TVM EP * fix logger name * return back to apache/tvm with VM fixes instead of local dev branch * hide customized tvm logger while issue is not resolved. fix tvm warning related to target_host * flake8 fix Co-authored-by: Valery Chernov <valery.chernov@deelvin.com>
36 lines
No EOL
903 B
CMake
36 lines
No EOL
903 B
CMake
if (onnxruntime_USE_TVM)
|
|
message(STATUS "onnxruntime_USE_TVM: Fetch tvm for TVM EP")
|
|
|
|
FetchContent_Declare(
|
|
tvm
|
|
GIT_REPOSITORY https://github.com/apache/tvm.git
|
|
GIT_TAG d62a364ba783afef92623ee531043ee8dbd43566
|
|
)
|
|
|
|
FetchContent_GetProperties(tvm)
|
|
if(NOT tvm_POPULATED)
|
|
FetchContent_Populate(tvm)
|
|
file(CREATE_LINK ${tvm_BINARY_DIR} ${tvm_SOURCE_DIR}/build SYMBOLIC)
|
|
endif()
|
|
|
|
set(tvm_INCLUDE_DIRS ${tvm_SOURCE_DIR}/include)
|
|
|
|
endif()
|
|
|
|
if (onnxruntime_USE_NUPHAR)
|
|
message(STATUS "onnxruntime_USE_NUPHAR: Fetch onnxruntime-tvm for NUPHAR EP")
|
|
|
|
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)
|
|
|
|
endif() |