diff --git a/.github/workflows/github_actions.yml b/.github/workflows/github_actions.yml index 701991ba7d..468401e9bc 100644 --- a/.github/workflows/github_actions.yml +++ b/.github/workflows/github_actions.yml @@ -498,9 +498,7 @@ jobs: - name: Get Instruction id: get-simd run: | - if [[ $(uname -s) == Linux ]]; then - echo "SIMD " $(lscpu | grep -i flags | sort | uniq) - fi + ./travis/cpuinfo.sh echo "simd=$(./travis/guess_simd.sh)" >> $GITHUB_ENV shell: bash - name: Setup cache diff --git a/travis/cpuinfo.sh b/travis/cpuinfo.sh new file mode 100755 index 0000000000..33f5b7b84d --- /dev/null +++ b/travis/cpuinfo.sh @@ -0,0 +1,10 @@ +#!/usr/bin/env bash +git clone https://github.com/pytorch/cpuinfo +cd cpuinfo +mkdir build && cd build +cmake .. +make -j3 +./isa-info +./cpu-info +cd ../.. +rm -rf cpuinfo