@@ -37,24 +37,6 @@ if [[ -z "${HOST_WORKSPACE:-}" ]]; then
3737 echo " Setting HOST_WORKSPACE to: $HOST_WORKSPACE "
3838fi
3939
40- if [[ -n " ${GITHUB_ACTIONS:- } " ]]; then
41- # Prepare mount points etc for getting artifacts in/out of the container.
42- source " $ci_dir /util/artifacts/common.sh"
43- # Note that these mounts use the runner (not the devcontainer) filesystem for
44- # source directories because of docker-out-of-docker quirks.
45- # The workflow-job GH actions make sure that they exist before running any
46- # scripts.
47- action_mounts=$( cat << EOF
48- --mount type=bind,source=${ARTIFACT_ARCHIVES} ,target=${ARTIFACT_ARCHIVES} \
49- --mount type=bind,source=${ARTIFACT_UPLOAD_STAGE} ,target=${ARTIFACT_UPLOAD_STAGE}
50- EOF
51- )
52-
53- else
54- # If not running in GitHub Actions, we don't need to set up artifact mounts.
55- action_mounts=" "
56- fi
57-
5840# pynvbench must be built in a container that can produce manylinux wheels,
5941# and has the CUDA toolkit installed. We use the rapidsai/ci-wheel image for this.
6042# We build separate wheels using separate containers for each CUDA version,
@@ -84,11 +66,7 @@ for ctk in 12 13; do
8466 docker run --rm -i \
8567 --workdir /workspace/python \
8668 --mount type=bind,source=${HOST_WORKSPACE} ,target=/workspace/ \
87- ${action_mounts} \
8869 --env py_version=${py_version} \
89- --env GITHUB_ACTIONS=${GITHUB_ACTIONS:- } \
90- --env GITHUB_RUN_ID=${GITHUB_RUN_ID:- } \
91- --env JOB_ID=${JOB_ID:- } \
9270 $image \
9371 /workspace/ci/build_pynvbench_wheel_for_cuda.sh
9472 # Prevent GHA runners from exhausting available storage with leftover images:
@@ -137,7 +115,7 @@ echo "Found CUDA 13 wheel: $cu13_wheel"
137115cu12_wheel=$( readlink -f " $cu12_wheel " )
138116cu13_wheel=$( readlink -f " $cu13_wheel " )
139117
140- # Merge the wheels manually (similar to CCCL)
118+ # Merge the wheels manually
141119mkdir -p wheelhouse_merged
142120cd wheelhouse_merged
143121
@@ -192,8 +170,3 @@ rm -rf wheelhouse_merged wheelhouse_final
192170
193171echo " Final wheels in wheelhouse:"
194172ls -la wheelhouse/
195-
196- if [[ -n " ${GITHUB_ACTIONS:- } " ]]; then
197- wheel_artifact_name=" $( ci/util/workflow/get_wheel_artifact_name.sh) "
198- ci/util/artifacts/upload.sh $wheel_artifact_name ' wheelhouse/.*'
199- fi
0 commit comments