Update Kokkos library in LAMMPS to v3.3.0
This commit is contained in:
@ -29,6 +29,19 @@ do
|
||||
KOKKOS_DEVICES="${KOKKOS_DEVICES},Cuda"
|
||||
CUDA_PATH="${key#*=}"
|
||||
;;
|
||||
--with-hip)
|
||||
KOKKOS_DEVICES="${KOKKOS_DEVICES},Hip"
|
||||
HIP_PATH_HIPCC=$(command -v hipcc)
|
||||
HIP_PATH=${HIP_PATH_HIPCC%/bin/hipcc}
|
||||
;;
|
||||
# Catch this before '--with-hip*'
|
||||
--with-hip-options*)
|
||||
KOKKOS_HIP_OPT="${key#*=}"
|
||||
;;
|
||||
--with-hip*)
|
||||
KOKKOS_DEVICES="${KOKKOS_DEVICES},Hip"
|
||||
HIP_PATH="${key#*=}"
|
||||
;;
|
||||
--with-openmp)
|
||||
KOKKOS_DEVICES="${KOKKOS_DEVICES},OpenMP"
|
||||
;;
|
||||
@ -161,9 +174,9 @@ do
|
||||
echo "--cxxflags=[FLAGS] Overwrite CXXFLAGS for library build and test"
|
||||
echo " build. This will still set certain required"
|
||||
echo " flags via KOKKOS_CXXFLAGS (such as -fopenmp,"
|
||||
echo " --std=c++11, etc.)."
|
||||
echo " --std=c++14, etc.)."
|
||||
echo "--cxxstandard=[FLAGS] Overwrite KOKKOS_CXX_STANDARD for library build and test"
|
||||
echo " c++11 (default), c++14, c++17, c++1y, c++1z, c++2a"
|
||||
echo " c++14 (default), c++17, c++1y, c++1z, c++2a"
|
||||
echo "--ldflags=[FLAGS] Overwrite LDFLAGS for library build and test"
|
||||
echo " build. This will still set certain required"
|
||||
echo " flags via KOKKOS_LDFLAGS (such as -fopenmp,"
|
||||
@ -221,6 +234,10 @@ elif
|
||||
[ ${#COMPILER} -eq 0 ] && [[ ${KOKKOS_DEVICES} =~ .*Cuda.* ]]; then
|
||||
COMPILER="${KOKKOS_PATH}/bin/nvcc_wrapper"
|
||||
KOKKOS_SETTINGS="${KOKKOS_SETTINGS} CXX=${COMPILER}"
|
||||
elif
|
||||
[ ${#COMPILER} -eq 0 ] && [[ ${KOKKOS_DEVICES} =~ .*Hip.* ]]; then
|
||||
COMPILER=hipcc
|
||||
KOKKOS_SETTINGS="${KOKKOS_SETTINGS} CXX=${COMPILER}"
|
||||
fi
|
||||
|
||||
if [ ${#KOKKOS_DEVICES} -gt 0 ]; then
|
||||
@ -239,6 +256,10 @@ if [ ${#CUDA_PATH} -gt 0 ]; then
|
||||
KOKKOS_SETTINGS="${KOKKOS_SETTINGS} CUDA_PATH=${CUDA_PATH}"
|
||||
fi
|
||||
|
||||
if [ ${#HIP_PATH} -gt 0 ]; then
|
||||
KOKKOS_SETTINGS="${KOKKOS_SETTINGS} HIP_PATH=${HIP_PATH}"
|
||||
fi
|
||||
|
||||
if [ ${#CXXFLAGS} -gt 0 ]; then
|
||||
KOKKOS_SETTINGS="${KOKKOS_SETTINGS} CXXFLAGS=\"${CXXFLAGS}\""
|
||||
fi
|
||||
|
||||
Reference in New Issue
Block a user