Files
TheRock/rocm-systems/projects/rccl/tools/ci/lib/ensure-python-yaml.sh
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

29 lines
1.3 KiB
Bash
Executable File

#!/usr/bin/env bash
# Ensure `python3` on PATH can `import yaml` for the RCCL / rccl-tests build.
#
# Some CMake steps shell out to python and `import yaml`; when PyYAML is missing,
# configure aborts with "ModuleNotFoundError: No module named 'yaml'". If it is
# not already importable, create a venv (reusing system site-packages) that
# pip-installs PyYAML and put it first on PATH.
#
# MUST be sourced (not executed) so the venv activation persists in the caller.
# Honors RCCL_CI_VENV to override the venv location.
if python3 -c 'import yaml' 2>/dev/null; then
echo "==> python yaml: already importable ($(command -v python3))"
else
_venv_dir="${RCCL_CI_VENV:-${RCCL_DEVICE_API_WORKDIR:?RCCL_DEVICE_API_WORKDIR must be set}/.ci-out/pyvenv}"
if [[ ! -x "${_venv_dir}/bin/python3" ]]; then
echo "==> python yaml: creating venv at ${_venv_dir}"
python3 -m venv --system-site-packages "${_venv_dir}"
fi
# shellcheck source=/dev/null
source "${_venv_dir}/bin/activate"
if ! python3 -c 'import yaml' 2>/dev/null; then
echo "==> python yaml: pip installing PyYAML into ${_venv_dir}"
python3 -m pip install --quiet --disable-pip-version-check PyYAML
fi
python3 -c 'import yaml, sys; print("==> python yaml:", yaml.__version__, "via", sys.executable)'
unset _venv_dir
fi