mirror of
https://github.com/nwchemgit/nwchem.git
synced 2026-07-28 22:25:48 -04:00
24 lines
870 B
Bash
Executable file
24 lines
870 B
Bash
Executable file
#!/usr/bin/env bash
|
|
UNAME_S=$(uname -s)
|
|
if [[ ${UNAME_S} == Linux ]]; then
|
|
CPU_FLAGS=$(cat /proc/cpuinfo | grep flags |tail -n 1)
|
|
CPU_FLAGS_2=$(cat /proc/cpuinfo | grep flags |tail -n 1)
|
|
elif [[ ${UNAME_S} == Darwin ]]; then
|
|
CPU_FLAGS=$(/usr/sbin/sysctl -n machdep.cpu.features)
|
|
CPU_FLAGS_2=$(/usr/sbin/sysctl -n machdep.cpu.leaf7_features)
|
|
fi
|
|
if [[ $(echo ${CPU_FLAGS} | tr 'A-Z' 'a-z'| awk ' /avx512f/{print "Y"}') == "Y" ]]; then
|
|
echo "avx512"
|
|
exit 0
|
|
elif [[ $(echo ${CPU_FLAGS_2} | tr 'A-Z' 'a-z'| awk ' /avx2/ {print "Y"}') == "Y" ]]; then
|
|
echo "avx2"
|
|
exit 0
|
|
elif [[ $(echo ${CPU_FLAGS} | tr 'A-Z' 'a-z'| awk ' /avx/ {print "Y"}') == "Y" ]]; then
|
|
echo "avx"
|
|
exit 0
|
|
elif [[ $(echo ${CPU_FLAGS} | tr 'A-Z' 'a-z'| awk ' /sse2/ {print "Y"}') == "Y" ]]; then
|
|
echo "sse2"
|
|
exit 0
|
|
else
|
|
echo "unknown"
|
|
fi
|