Skip to content

Commit

Permalink
use 'network', install 'ninja' only
Browse files Browse the repository at this point in the history
Co-authered-by: Rickard <[email protected]>
  • Loading branch information
wkpark committed Feb 8, 2024
1 parent f643669 commit bafe423
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/python-package.yml
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ jobs:
id: cuda-toolkit
with:
cuda: ${{ matrix.cuda_version }}
method: 'local'
method: 'network'
sub-packages: '["nvcc","cudart","cusparse","cublas","thrust","nvrtc_dev","cublas_dev","cusparse_dev"]'
linux-local-args: '["--toolkit"]'
use-github-cache: false
Expand All @@ -117,6 +117,7 @@ jobs:
set -ex
build_os=${{ matrix.os }}
build_arch=${{ matrix.arch }}
[[ "${{ matrix.os }}" = windows-* ]] && python3 -m pip install ninja
for NO_CUBLASLT in ON OFF; do
if [ ${build_os:0:6} == ubuntu ]; then
image=nvidia/cuda:${{ matrix.cuda_version }}-devel-ubuntu22.04
Expand All @@ -127,7 +128,6 @@ jobs:
&& cmake -DCOMPUTE_BACKEND=cuda -DNO_CUBLASLT=${NO_CUBLASLT} . \
&& cmake --build ."
else
python3 -m pip install cmake==3.27.9 ninja
cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DNO_CUBLASLT=${NO_CUBLASLT} -DCMAKE_BUILD_TYPE=Release -S .
cmake --build . --config Release
fi
Expand Down

0 comments on commit bafe423

Please sign in to comment.