wehub-resource-sync
35 lines
982 B
Bash
35 lines
982 B
Bash
#!/bin/bash
|
|
# Helper script to build dgl sparse libraries for PyTorch
|
|
set -e
|
|
|
|
mkdir -p build
|
|
mkdir -p $BINDIR/dgl_sparse
|
|
cd build
|
|
|
|
if [ $(uname) = 'Darwin' ]; then
|
|
CPSOURCE=*.dylib
|
|
else
|
|
CPSOURCE=*.so
|
|
fi
|
|
|
|
CMAKE_FLAGS="-DCUDA_TOOLKIT_ROOT_DIR=$CUDA_TOOLKIT_ROOT_DIR -DTORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST -DUSE_CUDA=$USE_CUDA -DEXTERNAL_DMLC_LIB_PATH=$EXTERNAL_DMLC_LIB_PATH"
|
|
# CMake passes in the list of directories separated by spaces. Here we replace them with semicolons.
|
|
CMAKE_FLAGS="$CMAKE_FLAGS -DDGL_INCLUDE_DIRS=${INCLUDEDIR// /;} -DDGL_BUILD_DIR=$BINDIR"
|
|
echo $CMAKE_FLAGS
|
|
|
|
if [ $# -eq 0 ]; then
|
|
$CMAKE_COMMAND $CMAKE_FLAGS ..
|
|
make -j
|
|
cp -v $CPSOURCE $BINDIR/dgl_sparse
|
|
else
|
|
for PYTHON_INTERP in $@; do
|
|
TORCH_VER=$($PYTHON_INTERP -c 'import torch; print(torch.__version__.split("+")[0])')
|
|
mkdir -p $TORCH_VER
|
|
cd $TORCH_VER
|
|
$CMAKE_COMMAND $CMAKE_FLAGS -DPYTHON_INTERP=$PYTHON_INTERP ../..
|
|
make -j
|
|
cp -v $CPSOURCE $BINDIR/dgl_sparse
|
|
cd ..
|
|
done
|
|
fi
|