guess_simd

travis/guess_simd.sh

@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+UNAME_S=$(uname -s)
+arch=`uname -m`
+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)
+    if [[ "$arch" == "x86_64" ]]; then
+	CPU_FLAGS_2=$(/usr/sbin/sysctl -n machdep.cpu.leaf7_features)
+    fi
+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
+elif [[ $arch == "arm64" || $arch == "aarch64" ]]; then
+    echo "arm64"
+    exit 0
+else
+    echo "unknown"
+fi