#!/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