# Makefile Example to deploy TVM modules. ROCM_PATH=/opt/rocm TVM_ROOT=$(shell cd ../..; pwd) NNVM_PATH=nnvm DMLC_CORE=${TVM_ROOT}/dmlc-core PKG_CFLAGS = -std=c++11 -O2 -fPIC\ -I${TVM_ROOT}/include\ -I${DMLC_CORE}/include\ -I${TVM_ROOT}/dlpack/include\ -I${ROCM_PATH}/include PKG_LDFLAGS = -L${ROCM_PATH}/lib -L${TVM_ROOT}/lib -ldl -lpthread -lhip_hcc -lMIOpen .PHONY: clean all all: lib/libtvm_runtime_rocm.so # Build rule for all in one TVM package library lib/libtvm_runtime_rocm.so: rocm_runtime_pack.cc @mkdir -p $(@D) $(CXX) $(PKG_CFLAGS) -shared -o $@ $(filter %.cc %.o %.a, $^) $(PKG_LDFLAGS)