dmlc--dgl
Quan (Andy) Gan
e35e860ae3
* cmake fixes for older systems * allow specification of cuda path * test script fixes to enable openmp & test * update minigun; disable minigun partial frontier compile
35 行
538 B
Bash
35 行
538 B
Bash
#!/bin/bash
|
|
set -e
|
|
|
|
if [ $# -ne 1 ]; then
|
|
echo "Device argument required, can be cpu or gpu"
|
|
exit -1
|
|
fi
|
|
|
|
CMAKE_VARS="-DBUILD_CPP_TEST=ON -DUSE_OPENMP=ON"
|
|
|
|
if [ "$1" == "gpu" ]; then
|
|
CMAKE_VARS="-DUSE_CUDA=ON $CMAKE_VARS"
|
|
fi
|
|
|
|
if [ -d build ]; then
|
|
rm -rf build
|
|
fi
|
|
mkdir build
|
|
|
|
rm -rf _download
|
|
|
|
pushd build
|
|
cmake $CMAKE_VARS ..
|
|
make -j4
|
|
popd
|
|
|
|
pushd python
|
|
rm -rf build *.egg-info dist
|
|
pip3 uninstall -y dgl
|
|
# test install
|
|
python3 setup.py install
|
|
# test inplace build (for cython)
|
|
python3 setup.py build_ext --inplace
|
|
popd
|