set correct environment variables for python wheels

PREDICT-EPFL · Nov 1, 2024 · afadc9c · afadc9c
1 parent f386017
commit afadc9c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -61,6 +61,7 @@ jobs:
       - name: Build wheels
         uses: pypa/[email protected]
         env:
+          CIBW_ENVIRONMENT: ${{ (matrix.target.arch == 'x86_64' || matrix.target.arch == 'amd64') && 'CMAKE_ARGS="-DBLASFEO_X64_DIR=/opt/blasfeo_x64 -DBLASFEO_X64_AVX2_DIR=/opt/blasfeo_x64_avx2 -DBLASFEO_X64_AVX512_DIR=/opt/blasfeo_x64_avx512"' || ((matrix.target.arch == 'arm64' || matrix.target.arch == 'aarch64') && 'CMAKE_ARGS="-DBLASFEO_ARM64_DIR=/opt/blasfeo_arm64"' || '') }}
           CIBW_ARCHS_LINUX: ${{ matrix.target.arch }}
           CIBW_BEFORE_ALL_LINUX: bash ci/cibw_before_all_linux.sh
           CIBW_BEFORE_ALL_WINDOWS: bash ci/cibw_before_all_windows.sh

diff --git a/ci/build_install_blasfeo.sh b/ci/build_install_blasfeo.sh
@@ -17,12 +17,12 @@ mkdir "build_$ARCH_SUFFIX"
 cd "build_$ARCH_SUFFIX"
 case "$(uname -sr)" in
   CYGWIN*|MINGW*|MINGW32*|MSYS*) # detect windows
-    cmake .. -G"Ninja" -DCMAKE_C_COMPILER=clang-cl -DCMAKE_BUILD_TYPE=Release -DBLASFEO_CROSSCOMPILING=ON -DTARGET="$BLASFEO_TARGET" -DBLASFEO_REF_API=OFF -DBLAS_API=OFF -DBLASFEO_EXAMPLES=OFF ${3:+-DCMAKE_INSTALL_PREFIX="$3"}
+    cmake .. -G"Ninja" -DCMAKE_C_COMPILER=clang-cl -DCMAKE_BUILD_TYPE=Release -DBLASFEO_CROSSCOMPILING=ON -DTARGET="$BLASFEO_TARGET" -DBLAS_API=OFF -DBLASFEO_EXAMPLES=OFF ${3:+-DCMAKE_INSTALL_PREFIX="$3"}
     cmake --build .
     cmake --install .
     ;;
   *) # other OS
-    cmake .. -DCMAKE_BUILD_TYPE=Release -DBLASFEO_CROSSCOMPILING=ON -DTARGET="$BLASFEO_TARGET" -DBLASFEO_REF_API=OFF -DBLAS_API=OFF -DBLASFEO_EXAMPLES=OFF ${3:+-DCMAKE_INSTALL_PREFIX="$3"}
+    cmake .. -DCMAKE_BUILD_TYPE=Release -DBLASFEO_CROSSCOMPILING=ON -DTARGET="$BLASFEO_TARGET" -DBLAS_API=OFF -DBLASFEO_EXAMPLES=OFF ${3:+-DCMAKE_INSTALL_PREFIX="$3"}
     cmake --build .
     if [ "$EUID" -ne 0 ] # check if already root
       then