onnx-mlir/.buildbot/p9.sh

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