Skip to content

Commit

Permalink
Add configure flags for ELPA build with MPI/OpenMP on Cray
Browse files Browse the repository at this point in the history
  • Loading branch information
mkrack committed Dec 30, 2021
1 parent ee8c129 commit f1ef200
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tools/toolchain/scripts/stage5/install_elpa.sh
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,7 @@ case "$with_elpa" in
if [ "$ENABLE_CRAY" = "__TRUE__" ]; then
# extra LDFLAGS needed
cray_ldflags="-dynamic"
cray_config_flags="--enable-allow-thread-limiting --without-threading-support-check-during-build"
fi

# ELPA-2017xxxx enables AVX2 by default, switch off if machine doesn't support it.
Expand All @@ -93,13 +94,13 @@ case "$with_elpa" in
grep '\bavx512cd\b' /proc/cpuinfo 1> /dev/null && AVX512_flags+=" -mavx512cd"
grep '\bavx512bw\b' /proc/cpuinfo 1> /dev/null && AVX512_flags+=" -mavx512bw"
grep '\bavx512v1\b' /proc/cpuinfo 1> /dev/null && AVX512_flags+=" -mavx512v1"
config_flags="--enable-avx=${has_AVX} --enable-avx2=${has_AVX2} --enable-avx512=${has_AVX512}"
config_flags="${cray_config_flags} --enable-avx=${has_AVX} --enable-avx2=${has_AVX2} --enable-avx512=${has_AVX512}"
else
AVX_flag=""
AVX512_flags=""
FMA_flag=""
SSE4_flag=""
config_flags="--disable-avx --disable-avx2 --disable-avx512 --disable-sse --disable-sse-assembly"
config_flags="${cray_config_flags} --disable-avx --disable-avx2 --disable-avx512 --disable-sse --disable-sse-assembly"
fi
for TARGET in "cpu" "nvidia"; do
[ "$TARGET" == "nvidia" ] && [ "$ENABLE_CUDA" != "__TRUE__" ] && continue
Expand Down

0 comments on commit f1ef200

Please sign in to comment.