Homec4science
Diffusion lammps (runner) Browse

lammps/lib/kokkos/core/src/build_common.sh
87fac223005brunner

Last Change
Raw File

build_common.sh
View Options

#!/bin/bash
#-----------------------------------------------------------------------------
# Shared portion of build script for the base Kokkos functionality
# Simple build script with options
#-----------------------------------------------------------------------------
if [ ! -d "${KOKKOS}" \
-o ! -d "${KOKKOS}/src" \
-o ! -d "${KOKKOS}/src/impl" \
-o ! -d "${KOKKOS}/src/Cuda" \
-o ! -d "${KOKKOS}/src/OpenMP" \
-o ! -d "${KOKKOS}/src/Threads" \
] ;
then
echo "Must set KOKKOS to the kokkos/core directory"
exit -1
fi
#-----------------------------------------------------------------------------
INC_PATH="-I${KOKKOS}/src"
INC_PATH="${INC_PATH} -I${KOKKOS}/../TPL"
#-----------------------------------------------------------------------------
while [ -n "${1}" ] ; do
ARG="${1}"
shift 1
case ${ARG} in
#----------- OPTIONS -----------
OPT | opt | O3 | -O3 ) OPTFLAGS="${OPTFLAGS} -O3" ;;
#-------------------------------
DBG | dbg | g | -g ) KOKKOS_EXPRESSION_CHECK=1 ;;
#-------------------------------
HWLOC | hwloc ) KOKKOS_HAVE_HWLOC=${1} ; shift 1 ;;
#-------------------------------
MPI | mpi )
KOKKOS_HAVE_MPI=${1} ; shift 1
CXX="${KOKKOS_HAVE_MPI}/bin/mpicxx"
LINK="${KOKKOS_HAVE_MPI}/bin/mpicxx"
INC_PATH="${INC_PATH} -I${KOKKOS_HAVE_MPI}/include"
;;
#-------------------------------
OMP | omp | OpenMP )
KOKKOS_HAVE_OPENMP=1
;;
#-------------------------------
CUDA | Cuda | cuda )
# CUDA_ARCH options: 20 30 35
CUDA_ARCH=${1} ; shift 1
#
# -x cu : process all files through the Cuda compiler as Cuda code.
# -lib -o : produce library
#
NVCC="nvcc -gencode arch=compute_${CUDA_ARCH},code=sm_${CUDA_ARCH}"
NVCC="${NVCC} -maxrregcount=64"
NVCC="${NVCC} -Xcompiler -Wall,-ansi"
NVCC="${NVCC} -lib -o libCuda.a -x cu"
NVCC_SOURCES="${NVCC_SOURCES} ${KOKKOS}/src/Cuda/*.cu"
LIB="${LIB} libCuda.a -L/usr/local/cuda/lib64 -lcudart -lcusparse"
;;#-------------------------------
CUDA_OSX | Cuda_OSX | cuda_osx )
# CUDA_ARCH options: 20 30 35
CUDA_ARCH=${1} ; shift 1
#
# -x cu : process all files through the Cuda compiler as Cuda code.
# -lib -o : produce library
#
NVCC="nvcc -gencode arch=compute_${CUDA_ARCH},code=sm_${CUDA_ARCH}"
NVCC="${NVCC} -maxrregcount=64"
NVCC="${NVCC} -Xcompiler -Wall,-ansi -Xcompiler -m64"
NVCC="${NVCC} -lib -o libCuda.a -x cu"
NVCC_SOURCES="${NVCC_SOURCES} ${KOKKOS}/src/Cuda/*.cu"
LIB="${LIB} libCuda.a -Xlinker -rpath -Xlinker /Developer/NVIDIA/CUDA-5.5/lib -L /Developer/NVIDIA/CUDA-5.5/lib -lcudart -lcusparse"
;;
#-------------------------------
GNU | gnu | g++ )
# Turn on lots of warnings and ansi compliance.
# The Trilinos build system requires '-pedantic'
#
CXX="g++ -Wall -Wextra -ansi -pedantic"
LINK="g++"
CXX="${CXX} -rdynamic -DENABLE_TRACEBACK"
LIB="${LIB} -ldl"
;;
#-------------------------------
GNU_OSX | gnu_osx | g++_osx )
# Turn on lots of warnings and ansi compliance.
# The Trilinos build system requires '-pedantic'
#
CXX="g++ -Wall -Wextra -ansi -pedantic -m64"
LINK="g++"
CXX="${CXX} -DENABLE_TRACEBACK"
LIB="${LIB} -ldl"
;;
#-------------------------------
INTEL | intel | icc | icpc )
# -xW = use SSE and SSE2 instructions
CXX="icpc -Wall"
LINK="icpc"
LIB="${LIB} -lstdc++"
;;
#-------------------------------
MPIINTEL | mpiintel | mpiicc | mpiicpc )
# -xW = use SSE and SSE2 instructions
CXX="mpiicpc -Wall"
LINK="mpiicpc"
LIB="${LIB} -lstdc++"
KOKKOS_HAVE_MPI=1
;;
#-------------------------------
MIC | mic )
CXX="icpc -mmic -ansi-alias -Wall"
LINK="icpc -mmic"
CXX="${CXX} -mGLOB_default_function_attrs=knc_stream_store_controls=2"
# CXX="${CXX} -vec-report6"
# CXX="${CXX} -guide-vec"
LIB="${LIB} -lstdc++"
COMPILE_MIC="on"
;;
#-------------------------------
MPIMIC | mpimic )
CXX="mpiicpc -mmic -ansi-alias -Wall"
LINK="mpiicpc -mmic"
KOKKOS_HAVE_MPI=1
CXX="${CXX} -mGLOB_default_function_attrs=knc_stream_store_controls=2"
# CXX="${CXX} -vec-report6"
# CXX="${CXX} -guide-vec"
LIB="${LIB} -lstdc++"
COMPILE_MIC="on"
;;
#-------------------------------
curie )
CXX="CC"
LINK="CC"
INC_PATH="${INC_PATH} -I/opt/cray/mpt/default/gni/mpich2-cray/74"
KOKKOS_HAVE_MPI=1
;;
#-------------------------------
MKL | mkl )
HAVE_MKL=${1} ; shift 1 ;
CXX_FLAGS="${CXX_FLAGS} -DKOKKOS_USE_MKL -I${HAVE_MKL}/include/"
ARCH="intel64"
if [ -n "${COMPILE_MIC}" ] ;
then
ARCH="mic"
fi
LIB="${LIB} -L${HAVE_MKL}/lib/${ARCH}/ -lmkl_intel_lp64 -lmkl_intel_thread -lmkl_core"
NVCC_FLAGS="${NVCC_FLAGS} -DKOKKOS_USE_MKL"
;;
#-------------------------------
CUSPARSE | cusparse )
CXX_FLAGS="${CXX_FLAGS} -DKOKKOS_USE_CUSPARSE"
NVCC_FLAGS="${NVCC_FLAGS} -DKOKKOS_USE_CUSPARSE"
LIB="${LIB} -lcusparse"
;;
#-------------------------------
AVX | avx )
CXX_FLAGS="${CXX_FLAGS} -mavx"
;;
#-------------------------------
*) echo 'unknown option: ' ${ARG} ; exit -1 ;;
esac
done
#-----------------------------------------------------------------------------
if [ -z "${CXX}" ] ;
then
echo "No C++ compiler selected"
exit -1
fi
if [ -n "${KOKKOS_HAVE_OPENMP}" ]
then
CXX="${CXX} -fopenmp"
CXX_SOURCES="${CXX_SOURCES} ${KOKKOS}/src/OpenMP/*.cpp"
fi
#-----------------------------------------------------------------------------
# Option for PTHREAD or WINTHREAD eventually
KOKKOS_HAVE_PTHREAD=1
if [ -n "${KOKKOS_HAVE_PTHREAD}" ] ;
then
LIB="${LIB} -lpthread"
fi
#-----------------------------------------------------------------------------
# Option for enabling the Serial device
KOKKOS_HAVE_SERIAL=1
#-----------------------------------------------------------------------------
# Attach options to compile lines
CXX="${CXX} ${OPTFLAGS}"
if [ -n "${NVCC}" ] ;
then
NVCC="${NVCC} ${OPTFLAGS}"
fi
#-----------------------------------------------------------------------------
CXX_SOURCES="${CXX_SOURCES} ${KOKKOS}/src/impl/*.cpp"
CXX_SOURCES="${CXX_SOURCES} ${KOKKOS}/src/Threads/*.cpp"
#-----------------------------------------------------------------------------
#
if [ -n "${KOKKOS_HAVE_HWLOC}" ] ;
then
if [ ! -d ${KOKKOS_HAVE_HWLOC} ] ;
then
echo "${KOKKOS_HAVE_HWLOC} does not exist"
exit 1
fi
echo "LD_LIBRARY_PATH must include ${KOKKOS_HAVE_HWLOC}/lib"
LIB="${LIB} -L${KOKKOS_HAVE_HWLOC}/lib -lhwloc"
INC_PATH="${INC_PATH} -I${KOKKOS_HAVE_HWLOC}/include"
fi
#-----------------------------------------------------------------------------
INC_PATH="${INC_PATH} -I."
CONFIG="KokkosCore_config.h"
rm -f ${CONFIG}
echo "#ifndef KOKKOS_CORE_CONFIG_H" >> ${CONFIG}
echo "#define KOKKOS_CORE_CONFIG_H" >> ${CONFIG}
if [ -n "${KOKKOS_HAVE_MPI}" ] ;
then
echo "#define KOKKOS_HAVE_MPI" >> ${CONFIG}
fi
if [ -n "${NVCC}" ] ;
then
echo "#define KOKKOS_HAVE_CUDA" >> ${CONFIG}
fi
if [ -n "${KOKKOS_HAVE_PTHREAD}" ] ;
then
echo "#define KOKKOS_HAVE_PTHREAD" >> ${CONFIG}
fi
if [ -n "${KOKKOS_HAVE_SERIAL}" ] ;
then
echo "#define KOKKOS_HAVE_SERIAL" >> ${CONFIG}
fi
if [ -n "${KOKKOS_HAVE_HWLOC}" ] ;
then
echo "#define KOKKOS_HAVE_HWLOC" >> ${CONFIG}
fi
if [ -n "${KOKKOS_HAVE_OPENMP}" ] ;
then
echo "#define KOKKOS_HAVE_OPENMP" >> ${CONFIG}
fi
if [ -n "${KOKKOS_EXPRESSION_CHECK}" ] ;
then
echo "#define KOKKOS_EXPRESSION_CHECK" >> ${CONFIG}
fi
echo "#endif" >> ${CONFIG}
#-----------------------------------------------------------------------------
c4science · Help