start_rpc_server.sh 212 Bytes
Newer Older
Tianqi Chen committed
1 2 3 4 5
#/bin/bash
PROJ_ROOT=$(realpath $(dirname "$0")/../..)
export PYTHONPATH=${PROJ_ROOT}/python:${PYTHONPATH}

python -m tvm.exec.rpc_server "$@" --load-library=${PROJ_ROOT}/apps/rocm_rpc/lib/libtvm_runtime_rocm.so