forked from lijiext/lammps
221 lines
8.2 KiB
Bash
Executable File
221 lines
8.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
KOKKOS_DEVICES=""
|
|
|
|
while [[ $# > 0 ]]
|
|
do
|
|
key="$1"
|
|
|
|
case $key in
|
|
--kokkos-path*)
|
|
KOKKOS_PATH="${key#*=}"
|
|
;;
|
|
--prefix*)
|
|
PREFIX="${key#*=}"
|
|
;;
|
|
--with-cuda)
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},Cuda"
|
|
CUDA_PATH_NVCC=`which nvcc`
|
|
CUDA_PATH=${CUDA_PATH_NVCC%/bin/nvcc}
|
|
;;
|
|
--with-cuda*)
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},Cuda"
|
|
CUDA_PATH="${key#*=}"
|
|
;;
|
|
--with-openmp)
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},OpenMP"
|
|
;;
|
|
--with-pthread)
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},Pthread"
|
|
;;
|
|
--with-serial)
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},Serial"
|
|
;;
|
|
--with-qthread*)
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},Qthread"
|
|
QTHREAD_PATH="${key#*=}"
|
|
;;
|
|
--with-devices*)
|
|
DEVICES="${key#*=}"
|
|
KOKKOS_DEVICES="${KOKKOS_DEVICES},${DEVICES}"
|
|
;;
|
|
--with-gtest*)
|
|
GTEST_PATH="${key#*=}"
|
|
;;
|
|
--with-hwloc*)
|
|
HWLOC_PATH="${key#*=}"
|
|
;;
|
|
--arch*)
|
|
KOKKOS_ARCH="${key#*=}"
|
|
;;
|
|
--cxxflags*)
|
|
CXXFLAGS="${key#*=}"
|
|
;;
|
|
--ldflags*)
|
|
LDFLAGS="${key#*=}"
|
|
;;
|
|
--debug|-dbg)
|
|
KOKKOS_DEBUG=yes
|
|
;;
|
|
--compiler*)
|
|
COMPILER="${key#*=}"
|
|
;;
|
|
--with-options*)
|
|
KOKKOS_OPT="${key#*=}"
|
|
;;
|
|
--help)
|
|
echo "Kokkos configure options:"
|
|
echo "--kokkos-path=/Path/To/Kokkos: Path to the Kokkos root directory"
|
|
echo ""
|
|
echo "--with-cuda[=/Path/To/Cuda]: enable Cuda and set path to Cuda Toolkit"
|
|
echo "--with-openmp: enable OpenMP backend"
|
|
echo "--with-pthread: enable Pthreads backend"
|
|
echo "--with-serial: enable Serial backend"
|
|
echo "--with-qthread=/Path/To/Qthread: enable Qthread backend"
|
|
echo "--with-devices: explicitly add a set of backends"
|
|
echo ""
|
|
echo "--arch=[OPTIONS]: set target architectures. Options are:"
|
|
echo " SNB = Intel Sandy/Ivy Bridge CPUs"
|
|
echo " HSW = Intel Haswell CPUs"
|
|
echo " KNC = Intel Knights Corner Xeon Phi"
|
|
echo " Kepler30 = NVIDIA Kepler generation CC 3.0"
|
|
echo " Kepler35 = NVIDIA Kepler generation CC 3.5"
|
|
echo " Kepler37 = NVIDIA Kepler generation CC 3.7"
|
|
echo " Maxwell50 = NVIDIA Maxwell generation CC 5.0"
|
|
echo " Power8 = IBM Power 8 CPUs"
|
|
echo ""
|
|
echo "--compiler=/Path/To/Compiler set the compiler"
|
|
echo "--debug,-dbg: enable Debugging"
|
|
echo "--cxxflags=[FLAGS] overwrite CXXFLAGS for library build and test build"
|
|
echo " This will still set certain required flags via"
|
|
echo " KOKKOS_CXXFLAGS (such as -fopenmp, --std=c++11, etc.)"
|
|
echo "--ldflags=[FLAGS] overwrite LDFLAGS for library build and test build"
|
|
echo " This will still set certain required flags via"
|
|
echo " KOKKOS_LDFLAGS (such as -fopenmp, -lpthread, etc.)"
|
|
echo "--with-gtest=/Path/To/Gtest: set path to gtest (used in unit and performance tests"
|
|
echo "--with-hwloc=/Path/To/Hwloc: set path to hwloc"
|
|
echo "--with-options=[OPTIONS]: additional options to Kokkos:"
|
|
echo " aggressive_vectorization = add ivdep on loops"
|
|
exit 0
|
|
;;
|
|
*)
|
|
# unknown option
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
# If KOKKOS_PATH undefined, assume parent dir of this
|
|
# script is the KOKKOS_PATH
|
|
if [ -z "$KOKKOS_PATH" ]; then
|
|
KOKKOS_PATH=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
|
|
else
|
|
# Ensure KOKKOS_PATH is abs path
|
|
KOKKOS_PATH=$( cd $KOKKOS_PATH && pwd )
|
|
fi
|
|
|
|
KOKKOS_OPTIONS="KOKKOS_PATH=${KOKKOS_PATH}"
|
|
|
|
if [ ${#COMPILER} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} CXX=${COMPILER}"
|
|
fi
|
|
if [ ${#PREFIX} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} PREFIX=${PREFIX}"
|
|
fi
|
|
if [ ${#KOKKOS_DEVICES} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} KOKKOS_DEVICES=${KOKKOS_DEVICES}"
|
|
fi
|
|
if [ ${#KOKKOS_ARCH} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} KOKKOS_ARCH=${KOKKOS_ARCH}"
|
|
fi
|
|
if [ ${#KOKKOS_DEBUG} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} KOKKOS_DEBUG=${KOKKOS_DEBUG}"
|
|
fi
|
|
if [ ${#CUDA_PATH} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} CUDA_PATH=${CUDA_PATH}"
|
|
fi
|
|
if [ ${#CXXFLAGS} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} CXXFLAGS=\"${CXXFLAGS}\""
|
|
fi
|
|
if [ ${#LDFLAGS} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} LDFLAGS=\"${LDFLAGS}\""
|
|
fi
|
|
if [ ${#GTEST_PATH} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} GTEST_PATH=${GTEST_PATH}"
|
|
else
|
|
GTEST_PATH=${KOKKOS_PATH}/tpls/gtest
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} GTEST_PATH=${GTEST_PATH}"
|
|
fi
|
|
if [ ${#HWLOC_PATH} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} HWLOC_PATH=${HWLOC_PATH} KOKKOS_USE_TPLS=hwloc"
|
|
fi
|
|
if [ ${#QTHREAD_PATH} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} QTHREAD_PATH=${QTHREAD_PATH}"
|
|
fi
|
|
if [ ${#KOKKOS_OPT} -gt 0 ]; then
|
|
KOKKOS_OPTIONS="${KOKKOS_OPTIONS} KOKKOS_OPTIONS=${KOKKOS_OPT}"
|
|
fi
|
|
mkdir core
|
|
mkdir core/unit_test
|
|
mkdir core/perf_test
|
|
mkdir containers
|
|
mkdir containers/unit_tests
|
|
mkdir containers/performance_tests
|
|
mkdir algorithms
|
|
mkdir algorithms/unit_tests
|
|
mkdir algorithms/performance_tests
|
|
mkdir example
|
|
mkdir example/fixture
|
|
mkdir example/feint
|
|
mkdir example/fenl
|
|
|
|
|
|
echo "Generating Makefile with options " ${KOKKOS_OPTIONS}
|
|
echo "KOKKOS_OPTIONS=${KOKKOS_OPTIONS}" > Makefile
|
|
echo "" >> Makefile
|
|
echo "lib:" >> Makefile
|
|
echo -e "\tcd core; \\" >> Makefile
|
|
echo -e "\tmake -j -j -f ${KOKKOS_PATH}/core/src/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo "" >> Makefile
|
|
echo "install: lib" >> Makefile
|
|
echo -e "\tcd core; \\" >> Makefile
|
|
echo -e "\tmake -j -f ${KOKKOS_PATH}/core/src/Makefile ${KOKKOS_OPTIONS} install" >> Makefile
|
|
echo "" >> Makefile
|
|
echo "build-test:" >> Makefile
|
|
echo -e "\tcd core/unit_test; \\" >> Makefile
|
|
echo -e "\tmake -j -f ${KOKKOS_PATH}/core/unit_test/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd core/perf_test; \\" >> Makefile
|
|
echo -e "\tmake -j -f ${KOKKOS_PATH}/core/perf_test/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd containers/unit_tests; \\" >> Makefile
|
|
echo -e "\tmake -j -f ${KOKKOS_PATH}/containers/unit_tests/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd containers/performance_tests; \\" >> Makefile
|
|
echo -e "\tmake -j -f ${KOKKOS_PATH}/containers/performance_tests/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd algorithms/unit_tests; \\" >> Makefile
|
|
echo -e "\tmake -j -f ${KOKKOS_PATH}/algorithms/unit_tests/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd example/fixture; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/example/fixture/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd example/feint; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/example/feint/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo -e "\tcd example/fenl; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/example/fenl/Makefile ${KOKKOS_OPTIONS}" >> Makefile
|
|
echo "" >> Makefile
|
|
echo "test: build-test" >> Makefile
|
|
echo -e "\tcd core/unit_test; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/core/unit_test/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd core/perf_test; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/core/perf_test/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd containers/unit_tests; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/containers/unit_tests/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd containers/performance_tests; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/containers/performance_tests/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd algorithms/unit_tests; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/algorithms/unit_tests/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd example/fixture; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/example/fixture/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd example/feint; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/example/feint/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
echo -e "\tcd example/fenl; \\" >> Makefile
|
|
echo -e "\tmake -f ${KOKKOS_PATH}/example/fenl/Makefile ${KOKKOS_OPTIONS} test" >> Makefile
|
|
|
|
|