"src/vscode:/vscode.git/clone" did not exist on "71ca10c6a41732a7806d40cd96ab16fd782dea69"
Unverified Commit d696558b authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix building script for tensoradapter on mac (#2472)


Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
parent a8ce0957
......@@ -113,7 +113,7 @@ install the Python binding for DGL.
mkdir build
cd build
cmake -DUSE_OPENMP=off ..
cmake -DUSE_OPENMP=off -DCMAKE_C_FLAGS='-DXBYAK_DONT_USE_MAP_JIT' -DCMAKE_CXX_FLAGS='-DXBYAK_DONT_USE_MAP_JIT' ..
make -j4
cd ../python
python setup.py install
......
......@@ -7,6 +7,12 @@ mkdir -p build
mkdir -p $BINDIR/tensoradapter/pytorch
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"
if [ $# -eq 0 ]; then
......@@ -18,6 +24,6 @@ else
rm -rf *
$CMAKE_COMMAND $CMAKE_FLAGS -DPYTHON_INTERP=$PYTHON_INTERP ..
make -j
cp -v *.so $BINDIR/tensoradapter/pytorch
cp -v $CPSOURCE $BINDIR/tensoradapter/pytorch
done
fi
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment