diff --git a/.github/workflows/concrete_python_release_gpu.yml b/.github/workflows/concrete_python_release_gpu.yml index 1f0ff05257..af12c5a4f0 100644 --- a/.github/workflows/concrete_python_release_gpu.yml +++ b/.github/workflows/concrete_python_release_gpu.yml @@ -126,6 +126,8 @@ jobs: push: needs: [build-linux-x86] runs-on: ubuntu-latest + outputs: + wheel_version: ${{ steps.version.outputs.wheel_version }} steps: - uses: actions/checkout@v3 @@ -155,10 +157,17 @@ jobs: # update indexes and invalidate cloudfront cache python .github/workflows/scripts/s3_update_html_indexes.py + - name: Output Wheel Version + id: version + run: | + export VERSION=`ls ./wheels/*manylinux* | head -n1 | cut -d "-" -f2` + echo "VERSION=$VERSION" + echo "wheel_version=$VERSION" >> "$GITHUB_OUTPUT" + test-gpu-wheel: needs: [push] uses: ./.github/workflows/start_slab.yml secrets: inherit with: command: concrete-python-test-gpu-wheel - user_inputs: "TODO" + user_inputs: "${{ needs.push.outputs.wheel_version }}" diff --git a/.github/workflows/concrete_python_test_gpu_wheel.yml b/.github/workflows/concrete_python_test_gpu_wheel.yml index 2f0d6f7891..af2374743a 100644 --- a/.github/workflows/concrete_python_test_gpu_wheel.yml +++ b/.github/workflows/concrete_python_test_gpu_wheel.yml @@ -39,9 +39,8 @@ jobs: with: python-version: ${{ matrix.python-version }} - # TODO: specify CP version - name: Install CP - run: pip install --pre --extra-index-url https://pypi.zama.ai/gpu/ concrete-python + run: pip install --pre --extra-index-url https://pypi.zama.ai/gpu/ "concrete-python==${{ env.CP_VERSION }}" - name: Checkout the repository uses: actions/checkout@v3