fix building script for tensoradapter on mac (dmlc#2472)

Co-authored-by: Jinjing Zhou <[email protected]>
feixian15 · Jan 1, 2021 · d696558 · d696558
1 parent a8ce095
commit d696558
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/docs/source/install/index.rst b/docs/source/install/index.rst
@@ -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

diff --git a/tensoradapter/pytorch/build.sh b/tensoradapter/pytorch/build.sh
@@ -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