Update entrypoint to support arm64 (triton-inference-server#3387)

* Update entrypoint to support arm64 * review edits
Weizhongjin · Sep 22, 2021 · bb7c6dc · bb7c6dc
1 parent 367dd9e
commit bb7c6dc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nvidia_entrypoint.sh b/nvidia_entrypoint.sh
@@ -69,7 +69,8 @@ else
   fi
 fi
 
-if ! cat /proc/cpuinfo | grep flags | sort -u | grep avx >& /dev/null; then
+# avx instructions are only available on x86, so don't check for aarch64
+if ! cat /proc/cpuinfo | grep flags | sort -u | grep avx >& /dev/null && [[ "aarch64" != $(uname -m) ]] ; then
   echo
   echo "ERROR: This container was built for CPUs supporting at least the AVX instruction set, but"
   echo "       the CPU detected was $(cat /proc/cpuinfo |grep "model name" | sed 's/^.*: //' | sort -u), which does not report"