build_dgl_asv.sh

#!/bin/bash

set -e

. /opt/conda/etc/profile.d/conda.sh

# Default building only with cpu
DEVICE=${DGL_BENCH_DEVICE:-cpu}

# build
if [[ $DEVICE == "cpu" ]]; then
    CMAKE_VARS=""
else
    CMAKE_VARS="-DUSE_CUDA=ON"
fi
mkdir -p build
pushd build
cmake $CMAKE_VARS ..
make -j
popd