mirror of
https://github.com/elder-plinius/OBLITERATUS.git
synced 2026-08-18 00:47:23 +02:00
ci: make CUDA conditional lane GPU-capable
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user