fix github workflow for amd cpu

block-hczhai · Nov 3, 2023 · e770aca · e770aca
1 parent b86276b
commit e770aca
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -65,6 +65,21 @@ jobs:
  sudo make
  sudo cp lib/*.a /usr/lib
 
+ - name: fix mkl for amd cpu
+ if: matrix.os == 'ubuntu-20.04'
+ env:
+ CC: gcc-9
+ CXX: g++-9
+ run: |
+ lscpu
+ export CPUTYPE=$(lscpu | grep 'Vendor ID' | awk '{print $3}')
+ echo $CPUTYPE
+ if [ "$CPUTYPE" = "AuthenticAMD" ]; then
+ echo "int mkl_serv_intel_cpu_true() { return 1; }" > fixcpu.c
+ $CC -shared -fPIC -o libfixcpu.so fixcpu.c
+ export LD_PRELOAD=$PWD/libfixcpu.so
+ fi
+
  - name: build gtest (macos)
  if: matrix.os == 'macos-12'
  run: |