Skip to content

Commit

Permalink
fixup scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
brandon-b-miller committed Jan 5, 2024
1 parent 56a6d74 commit 01b4140
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
4 changes: 1 addition & 3 deletions ci/build_wheel.sh
Original file line number Diff line number Diff line change
@@ -1,16 +1,14 @@
#!/bin/bash
# Copyright (c) 2023, NVIDIA CORPORATION

set -euo pipefail

rapids-logger "Install CUDA Toolkit"
yum update
yum install -y epel-release
yum install -y nvidia-driver-latest-dkms
yum install -y cuda-toolkit-12-3

rapids-logger "Check conda environment"
conda list

rapids-logger "Build wheel"
mkdir -p ./wheel-build
pip wheel . --wheel-dir=./wheel-build -vvv
Expand Down
2 changes: 1 addition & 1 deletion ci/test_wheel.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#!/bin/bash
# Copyright (c) 2023, NVIDIA CORPORATION

set -e
set -euo pipefail

rapids-logger "Download Wheel"
RAPIDS_PY_WHEEL_NAME="pynvjitlink-cu12" rapids-download-wheels-from-s3 ./wheel-build/
Expand Down

0 comments on commit 01b4140

Please sign in to comment.