tools/cpp_build/build_caffe2.sh

#!/usr/bin/env bash

set -ex

SCRIPTPATH="$( cd "$(dirname "$0")" ; pwd -P )"

pushd $SCRIPTPATH
source ./build_common.sh

echo "Building Caffe2"

mkdir -p $CAFFE2_BUILDPATH
pushd $CAFFE2_BUILDPATH

cmake -DUSE_CUDA:BOOL=$USE_CUDA \
      -DBUILD_TORCH=ON \
      -DUSE_OPENMP:BOOL=${USE_OPENMP:ON} \
      -DBUILD_CAFFE2=OFF \
      -DBUILD_ATEN=ON \
      -DBUILD_PYTHON=OFF \
      -DBUILD_BINARY=OFF \
      -DBUILD_SHARED_LIBS=ON \
      -DONNX_NAMESPACE=$ONNX_NAMESPACE \
      -DCMAKE_BUILD_TYPE:STRING=$BUILD_TYPE \
      -DCMAKE_INSTALL_PREFIX:STRING=$INSTALL_PREFIX \
      -DCMAKE_INSTALL_MESSAGE=NEVER \
      -DCMAKE_EXPORT_COMPILE_COMMANDS:BOOL=ON \
      -G "$GENERATE" \
      $PYTORCHPATH/
$MAKE -j "$JOBS" install

popd
popd