TensorRT-LLMs/examples/cpp_library/build.sh
2023-09-20 00:29:41 -07:00

26 lines
554 B
Bash
Executable File

#!/usr/bin/env bash
BUILD_DIR="build"
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
rm -rf ${BUILD_DIR} && mkdir -p ${BUILD_DIR}
pushd ${BUILD_DIR}
cmake \
-DCMAKE_BUILD_TYPE=Release \
..
make -j"$(grep -c ^processor /proc/cpuinfo)"
export LD_LIBRARY_PATH="${SCRIPT_DIR}:${LD_LIBRARY_PATH}"
# Test Lib
echo
echo "--------------------------------------------------------------------"
./trt_llm_plugins_cpp_load_example
echo "--------------------------------------------------------------------"
echo
popd