Try fixing CI (#252)

* Try fixing CI * Update build.yml * Update tag.yml * Update gpu_pipeline.yml * Update pipeline.yml * Update build.yml * Update tag.yml
EnzymeAD · Jan 20, 2025 · 55d9211 · 55d9211
1 parent 4d27677
commit 55d9211
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 8 deletions.
diff --git a/.buildkite/gpu_pipeline.yml b/.buildkite/gpu_pipeline.yml
@@ -40,8 +40,8 @@ steps:
       echo "--- :python: Test"
 
       HERMETIC_PYTHON_VERSION="3.11" .local/bin/bazel --output_user_root=`pwd`/.baztmp run //builddeps:requirements.update
-      HERMETIC_PYTHON_VERSION="3.11" .local/bin/bazel --output_user_root=`pwd`/.baztmp build --test_output=errors //:wheel
-      HERMETIC_PYTHON_VERSION="3.11" .local/bin/bazel --output_user_root=`pwd`/.baztmp test --test_output=errors //test/...
+      HERMETIC_PYTHON_VERSION="3.11" .local/bin/bazel --output_user_root=`pwd`/.baztmp build --define=xnn_enable_avxvnniint8=false --define=xnn_enable_avxvnni=false --test_output=errors //:wheel
+      HERMETIC_PYTHON_VERSION="3.11" .local/bin/bazel --output_user_root=`pwd`/.baztmp test --define=xnn_enable_avxvnniint8=false --define=xnn_enable_avxvnni=false --test_output=errors //test/...
 
     artifact_paths:
       - "bazel-out/*/testlogs/test/llama/test.log"

diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -31,12 +31,18 @@ steps:
       echo "openssl md5 | cut -d' ' -f2" > .local/bin/md5
       chmod +x .local/bin/md5
 
+      BAZEL_BUILD_FLAGS=()
+      
       if [ "{{matrix.os}}" == "macos" ]; then
         if [ ! -f ".local/bin/bazel" ]; then
           curl -fLO "https://github.com/bazelbuild/bazelisk/releases/download/v1.19.0/bazelisk-darwin"
           mv bazelisk-darwin .local/bin/bazel
           chmod +x .local/bin/bazel
         fi
+        if [ "{{matrix.arch}}" == "aarch64" ]; then
+        else
+          BAZEL_BUILD_FLAGS+=(--define=xnn_enable_avxvnniint8=false)
+        fi
       elif [ "{{matrix.os}}" == "linux" ]; then
         mv bazel* .local/bin/bazel
         chmod +x .local/bin/bazel
@@ -50,9 +56,9 @@ steps:
       fi
       mkdir -p .baztmp
       HERMETIC_PYTHON_VERSION={{matrix.python}} bazel --output_user_root=`pwd`/.baztmp run //builddeps:requirements.update
-      HERMETIC_PYTHON_VERSION={{matrix.python}} bazel --output_user_root=`pwd`/.baztmp test --test_output=errors //test/...
+      HERMETIC_PYTHON_VERSION={{matrix.python}} bazel --output_user_root=`pwd`/.baztmp test ${BAZEL_BUILD_FLAGS[@]} --test_output=errors //test/...
       rm -f bazel-bin/*.whl
-      HERMETIC_PYTHON_VERSION={{matrix.python}} bazel --output_user_root=`pwd`/.baztmp build :wheel
+      HERMETIC_PYTHON_VERSION={{matrix.python}} bazel --output_user_root=`pwd`/.baztmp build ${BAZEL_BUILD_FLAGS[@]} :wheel
       cp bazel-bin/*.whl .
     artifact_paths:
       - "*.whl"

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -45,12 +45,12 @@ jobs:
 
       - name: test
         run: |
-          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel test --test_output=errors ...
+          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel test --define=xnn_enable_avxvnniint8=false --define=xnn_enable_avx512fp16=false --test_output=errors ...
           
       - name: Build Wheel
         run: |
           sudo rm bazel-bin/*.whl || echo
-          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel build :wheel
+          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel build --define=xnn_enable_avxvnniint8=false --define=xnn_enable_avx512fp16=false :wheel
           cp bazel-bin/*.whl .
 
       - name: Upload Build

diff --git a/.github/workflows/tag.yml b/.github/workflows/tag.yml
@@ -43,9 +43,9 @@ jobs:
           bazelisk-version: 1.x
       # - run: sudo find ~/.cache/bazel ~/.cache/bazelisk -iname "*.whl" -exec rm {} \;
       - run: |
-          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel build -c opt @llvm-project//llvm:FileCheck
+          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel build --define=xnn_enable_avxvnniint8=false --define=xnn_enable_avx512fp16=false -c opt @llvm-project//llvm:FileCheck
           sudo rm bazel-bin/*.whl || echo
-          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel build -c opt :wheel
+          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel build --define=xnn_enable_avxvnniint8=false --define=xnn_enable_avx512fp16=false -c opt :wheel
       - env:
           TWINE_USERNAME: __token__
           TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}