Skip to content

Commit 7ed6a01

Browse files
authored
Merge pull request #14 from rickardp/nocublaslt
Build nocublaslt versions
2 parents e2e8f87 + 2ba8be3 commit 7ed6a01

File tree

2 files changed

+14
-171
lines changed

2 files changed

+14
-171
lines changed

.github/workflows/cmake.yml

Lines changed: 0 additions & 159 deletions
This file was deleted.

.github/workflows/python-package.yml

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -119,18 +119,20 @@ jobs:
119119
set -ex
120120
build_os=${{ matrix.os }}
121121
build_arch=${{ matrix.arch }}
122-
if [ ${build_os:0:6} == ubuntu ]; then
123-
image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu22.04
124-
echo "Using image $image"
125-
docker run --platform linux/$build_arch -i -w /src -v $PWD:/src $image sh -c \
126-
"apt-get update \
127-
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
128-
&& cmake -DCOMPUTE_BACKEND=cuda . \
129-
&& make"
130-
else
131-
cmake -DCOMPUTE_BACKEND=cuda .
132-
pwsh -Command "msbuild bitsandbytes.vcxproj /property:Configuration=Release"
133-
fi
122+
for NO_CUBLASLT in ON OFF; do
123+
if [ ${build_os:0:6} == ubuntu ]; then
124+
image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu22.04
125+
echo "Using image $image"
126+
docker run --platform linux/$build_arch -i -w /src -v $PWD:/src $image sh -c \
127+
"apt-get update \
128+
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
129+
&& cmake -DCOMPUTE_BACKEND=cuda -DNO_CUBLASLT=${NO_CUBLASLT} . \
130+
&& make"
131+
else
132+
cmake -DCOMPUTE_BACKEND=cuda -DNO_CUBLASLT=${NO_CUBLASLT} .
133+
pwsh -Command "msbuild bitsandbytes.vcxproj /property:Configuration=Release"
134+
fi
135+
done
134136
mkdir -p output/${{ matrix.os }}/${{ matrix.arch }}
135137
( shopt -s nullglob && cp bitsandbytes/*.{so,dylib,dll} output/${{ matrix.os }}/${{ matrix.arch }}/ )
136138
- name: Upload build artifact

0 commit comments

Comments
 (0)