ci: make CUDA conditional lane GPU-capable

This commit is contained in:
Joseph Magly
2026-08-16 09:23:22 -04:00
parent baf5331730
commit fa233fd8c9
4 changed files with 42 additions and 5 deletions
+9
View File
@@ -240,6 +240,15 @@ jobs:
python -m pip install "uv==${UV_VERSION}"
UV_PROJECT_ENVIRONMENT="$CONDITIONAL_ENV" \
uv sync --locked --no-default-groups --extra dev --no-editable
CUDA_TORCH_VERSION="$("$CONDITIONAL_ENV/bin/python" -c \
'import torch; print(torch.__version__.split("+", 1)[0])')"
UV_TORCH_BACKEND=cu130 uv pip install \
--python "$CONDITIONAL_ENV/bin/python" \
--reinstall-package torch \
"torch==$CUDA_TORCH_VERSION"
"$CONDITIONAL_ENV/bin/python" -c \
'import torch; assert torch.version.cuda is not None, torch.__version__'
uv pip check --python "$CONDITIONAL_ENV/bin/python"
- name: Run CUDA placement and operation probe
run: >-
"$CONDITIONAL_ENV/bin/python" scripts/run_conditional_gate.py cuda-runtime