diff --git a/llvm/projects/hpvm-tensor-rt/bin/setup_jetson.sh b/llvm/projects/hpvm-tensor-rt/bin/setup_jetson.sh index 2ad4d5bc765e23841b87ebdcb778295456c2b380..b288ccfe43c577f9ad14c4eb16284539ae5682ea 100644 --- a/llvm/projects/hpvm-tensor-rt/bin/setup_jetson.sh +++ b/llvm/projects/hpvm-tensor-rt/bin/setup_jetson.sh @@ -1,3 +1,8 @@ export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-9.0/targets/aarch64-linux/lib/ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/targets/aarch64-linux/lib/ +export CUDNN_PATH=/usr/local/cuda-9.0/ +export CUDA_INCLUDE_PATH=${CUDNN_PATH}/include + +export LLVM_BUILD_ROOT=/home/nvidia/Gitlab/hpvm/build/ +export LLVM_SRC_ROOT=/home/nvidia/Gitlab/hpvm/llvm/