|
|
|
@ -59,9 +59,9 @@ function init() {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
function cmake_base() {
|
|
|
|
|
# build script will not fail if *.deb does not exist
|
|
|
|
|
# Build script will not fail if *.deb does not exist
|
|
|
|
|
rm *.deb 2>/dev/null || true
|
|
|
|
|
# delete previous built whl packages
|
|
|
|
|
# Delete previous built whl packages
|
|
|
|
|
rm -rf python/dist 2>/dev/null || true
|
|
|
|
|
|
|
|
|
|
# Support build for all python versions, currently
|
|
|
|
@ -199,9 +199,7 @@ function cmake_base() {
|
|
|
|
|
-DWITH_DISTRIBUTE=${distibuted_flag}
|
|
|
|
|
-DWITH_MKL=${WITH_MKL:-ON}
|
|
|
|
|
-DWITH_AVX=${WITH_AVX:-OFF}
|
|
|
|
|
-DWITH_GOLANG=${WITH_GOLANG:-OFF}
|
|
|
|
|
-DCUDA_ARCH_NAME=${CUDA_ARCH_NAME:-All}
|
|
|
|
|
-DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
|
|
|
|
|
-DWITH_PYTHON=${WITH_PYTHON:-ON}
|
|
|
|
|
-DCUDNN_ROOT=/usr/
|
|
|
|
|
-DWITH_TESTING=${WITH_TESTING:-ON}
|
|
|
|
@ -231,9 +229,7 @@ EOF
|
|
|
|
|
-DWITH_MKL=${WITH_MKL:-ON} \
|
|
|
|
|
-DWITH_AVX=${WITH_AVX:-OFF} \
|
|
|
|
|
-DNOAVX_CORE_FILE=${NOAVX_CORE_FILE:-""} \
|
|
|
|
|
-DWITH_GOLANG=${WITH_GOLANG:-OFF} \
|
|
|
|
|
-DCUDA_ARCH_NAME=${CUDA_ARCH_NAME:-All} \
|
|
|
|
|
-DCUDA_ARCH_BIN=${CUDA_ARCH_BIN} \
|
|
|
|
|
-DWITH_PYTHON=${WITH_PYTHON:-ON} \
|
|
|
|
|
-DCUDNN_ROOT=/usr/ \
|
|
|
|
|
-DWITH_TESTING=${WITH_TESTING:-ON} \
|
|
|
|
@ -1080,7 +1076,7 @@ EOF
|
|
|
|
|
if [[ "$1" != "" ]]; then
|
|
|
|
|
parallel_number=$1
|
|
|
|
|
fi
|
|
|
|
|
cmake .. -DWITH_DISTRIBUTE=OFF -DON_INFER=ON -DCUDA_ARCH_NAME=${CUDA_ARCH_NAME:-Auto} -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
|
|
|
|
|
cmake .. -DWITH_DISTRIBUTE=OFF -DON_INFER=ON -DCUDA_ARCH_NAME=${CUDA_ARCH_NAME:-Auto}
|
|
|
|
|
|
|
|
|
|
make -j ${parallel_number} fluid_lib_dist
|
|
|
|
|
make -j ${parallel_number} inference_lib_dist
|
|
|
|
|