Merge pull request #14 from rickardp/nocublaslt

Build nocublaslt versions
bitsandbytes-foundation · Feb 3, 2024 · 7ed6a01 · 7ed6a01
2 parents e2e8f87 + 2ba8be3
commit 7ed6a01
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 171 deletions.
diff --git a/.github/workflows/cmake.yml b/.github/workflows/cmake.yml
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -119,18 +119,20 @@ jobs:
         set -ex
         build_os=${{ matrix.os }}
         build_arch=${{ matrix.arch }}
-        if [ ${build_os:0:6} == ubuntu ]; then 
-          image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu22.04
-          echo "Using image $image"
-          docker run --platform linux/$build_arch -i -w /src -v $PWD:/src $image sh -c \
-            "apt-get update \
-            && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
-            && cmake -DCOMPUTE_BACKEND=cuda . \
-            && make"
-        else
-          cmake -DCOMPUTE_BACKEND=cuda .
-          pwsh -Command "msbuild bitsandbytes.vcxproj /property:Configuration=Release"
-        fi
+        for NO_CUBLASLT in ON OFF; do
+          if [ ${build_os:0:6} == ubuntu ]; then 
+            image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu22.04
+            echo "Using image $image"
+            docker run --platform linux/$build_arch -i -w /src -v $PWD:/src $image sh -c \
+              "apt-get update \
+              && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
+              && cmake -DCOMPUTE_BACKEND=cuda -DNO_CUBLASLT=${NO_CUBLASLT} . \
+              && make"
+          else
+            cmake -DCOMPUTE_BACKEND=cuda -DNO_CUBLASLT=${NO_CUBLASLT} .
+            pwsh -Command "msbuild bitsandbytes.vcxproj /property:Configuration=Release"
+          fi
+        done
         mkdir -p output/${{ matrix.os }}/${{ matrix.arch }}
         ( shopt -s nullglob && cp bitsandbytes/*.{so,dylib,dll} output/${{ matrix.os }}/${{ matrix.arch }}/ )
     - name: Upload build artifact