check_cpu.sh

set -e
QUERY="$1"

check_mac() {
    case "$1" in
      avx) sysctl -n machdep.cpu.features | grep -i avx >/dev/null;;
      avx2) sysctl -n machdep.cpu.features | grep -i avx2 >/dev/null;;
      amd) sysctl -n machdep.cpu.vendor | grep -i amd >/dev/null;;
      amdnew) sysctl -n machdep.cpu.vendor | grep -i amd && test `sysctl -n machdep.cpu.family` -ge 23;;
      intel) sysctl -n machdep.cpu.vendor | grep -i intel >/dev/null;;
      sse2) sysctl -n machdep.cpu.features | grep -i sse2 >/dev/null;;
      ssse3) sysctl -n machdep.cpu.features | grep -i ssse3 >/dev/null;;
      avx512f) sysctl -n machdep.cpu.features | grep -i avx512f >/dev/null;;
      xop) sysctl -n machdep.cpu.features | grep -i xop >/dev/null;;
      *) echo "UNRECOGNISED CHECK $QUERY"; exit 1; ;;
    esac
}

check_linux() {
    case "$1" in
      avx) grep avx /proc/cpuinfo >/dev/null;;
      avx2) grep avx2 /proc/cpuinfo >/dev/null;;
      amd) grep -i amd /proc/cpuinfo >/dev/null;;
      amdnew) grep -i amd /proc/cpuinfo >/dev/null && test `awk '/cpu family/ && $NF~/^[0-9]*$/ {print $NF}' /proc/cpuinfo | head -n1` -ge 23 >/dev/null;;
      intel) grep -i intel /proc/cpuinfo >/dev/null;;
      sse2) grep sse2 /proc/cpuinfo >/dev/null;;
      ssse3) grep ssse3 /proc/cpuinfo >/dev/null;;
      avx512f) grep avx512f /proc/cpuinfo >/dev/null;;
      xop) grep xop /proc/cpuinfo >/dev/null;;
      *) echo "UNRECOGNISED CHECK $QUERY"; exit 1; ;;
    esac
}

case "$(uname -a)" in
  Darwin*) check_mac "$QUERY" ;;
  *) check_linux "$QUERY" ;;
esac