Skip to content

Instantly share code, notes, and snippets.

@cyrusbehr
Last active December 3, 2019 17:38
Show Gist options
  • Save cyrusbehr/2b5db3fafa1f4bbf9be01083d7f5e3c2 to your computer and use it in GitHub Desktop.
Save cyrusbehr/2b5db3fafa1f4bbf9be01083d7f5e3c2 to your computer and use it in GitHub Desktop.
Performing inference with MXNet model converted to TVM
cmake_minimum_required(VERSION 3.14)
project(tvm-inference)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14 -fPIC -ffast-math -Ofast -ldl -lpthread")
add_executable(tvm_test tvm_runtime_pack.cc main.cpp)
set(HOME_TVM /path/to/tvm/dir)
find_package(OpenCV REQUIRED)
target_include_directories(
tvm_test
PRIVATE
${OpenCV_INCLUDE_DIRS}
${HOME_TVM}/include
${HOME_TVM}/3rdparty/dmlc-core/include
${HOME_TVM}/3rdparty/dlpack/include
)
target_link_libraries(
tvm_test
PRIVATE
${OpenCV_LIBS}
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment