48 lines
1.6 KiB
Bash
48 lines
1.6 KiB
Bash
#!/bin/bash
|
|
|
|
# Check for required env variables ONNF_DEP_DIR, LLVM_PROJECT_ROOT
|
|
if [[ -z "${ONNF_DEP_DIR}" ]]; then
|
|
echo "ONNF_DEP_DIR env var is missing."
|
|
exit 1
|
|
fi
|
|
if [[ -z "${LLVM_PROJECT_ROOT}" ]]; then
|
|
echo "LLVM_PROJECT_ROOT env var is missing."
|
|
exit 1
|
|
fi
|
|
|
|
# Set up env variables to expose onnf dependencies:
|
|
export PATH=$ONNF_DEP_DIR/bin:$PATH
|
|
export LD_LIBRARY_PATH=$ONNF_DEP_DIR/lib:$ONNF_DEP_DIR/lib64:
|
|
export CPATH=$ONNF_DEP_DIR/include:$CPATH
|
|
export PATH=$ONNF_DEP_DIR/bin:$PATH
|
|
|
|
# Set up mock installation path within current workspace:
|
|
export INSTALL_PATH=$WORKSPACE/INSTALL_PATH
|
|
mkdir -p "$INSTALL_PATH"
|
|
export PATH=$INSTALL_PATH/bin:$PATH
|
|
export LD_LIBRARY_PATH=$INSTALL_PATH/lib:$INSTALL_PATH/lib64:$LD_LIBRARY_PATH
|
|
export CPATH=$INSTALL_PATH/include:$CPATH
|
|
|
|
# Create virtual environment specific to the current build instance:
|
|
conda create -n onnf_conda_workspace_"${BUILD_NUMBER}" python=3.7 numpy
|
|
source activate onnf_conda_workspace_"${BUILD_NUMBER}"
|
|
|
|
# Create build directory and generate make files:
|
|
mkdir build && cd build
|
|
CC=$ONNF_DEP_DIR/bin/gcc \
|
|
CXX=$ONNF_DEP_DIR/bin/g++ \
|
|
BOOST_ROOT=$ONNF_DEP_DIR \
|
|
LLVM_SRC=$LLVM_PROJECT_ROOT/llvm \
|
|
LLVM_BUILD=$LLVM_PROJECT_ROOT/build \
|
|
cmake3 -DONNF_ENABLE_MODEL_TEST_CPP=ON \
|
|
-DONNF_ENABLE_BENCHMARK=ON \
|
|
-DCMAKE_INSTALL_PREFIX="$INSTALL_PATH" \
|
|
..
|
|
|
|
# Build and test:
|
|
make -j "$(nproc)" install
|
|
OMP_NUM_THREADS=20 OMP_THREAD_LIMIT=20 ctest3 -j "$(nproc)"
|
|
|
|
# Run lit+FileCheck tests:
|
|
make check-mlir-lit
|