diff --git a/.github/workflows/test_gpu_cu11.yml b/.github/workflows/test_gpu_cu11.yml index a78884058e..ebdd6e6fc5 100644 --- a/.github/workflows/test_gpu_cu11.yml +++ b/.github/workflows/test_gpu_cu11.yml @@ -7,10 +7,10 @@ on: env: CI_CUDA_ARCH: 86 - CUDA_VERSION_MAJOR: 11 - CUDA_VERSION_MINOR: 8 COVERAGE_FLAGS: "--cov=pennylane_lightning --cov-report=term-missing --cov-report=xml:./coverage.xml --no-flaky-report -p no:warnings --tb=native" GCC_VERSION: 11 + CUDA_VERSION: 11.8 + CUDA_MAJOR_VERSION: 11 TF_VERSION: 2.10.0 TORCH_VERSION: 1.11.0+cpu @@ -36,7 +36,7 @@ jobs: run: | source /etc/profile.d/modules.sh module use /opt/modules - module load cuda/$CUDA_VERSION_MAJOR.$CUDA_VERSION_MINOR + module load cuda/$CUDA_VERSION echo "${PATH}" >> $GITHUB_PATH echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV nvcc --version @@ -57,7 +57,7 @@ jobs: run: | source /etc/profile.d/modules.sh module use /opt/modules - module load cuda/11.8 + module load cuda/$CUDA_VERSION echo "${PATH}" >> $GITHUB_PATH echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV nvcc --version @@ -105,7 +105,7 @@ jobs: - name: Install required packages run: | - python -m pip install ninja cmake custatevec-cu11 + python -m pip install ninja cmake custatevec-cu$CUDA_MAJOR_VERSION - name: Install Latest PennyLane if: inputs.pennylane-version == 'latest' @@ -160,7 +160,7 @@ jobs: run: | source /etc/profile.d/modules.sh module use /opt/modules - module load cuda/11.8 + module load cuda/$CUDA_VERSION echo "${PATH}" >> $GITHUB_PATH echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV nvcc --version @@ -214,7 +214,7 @@ jobs: - name: Install required packages run: | python -m pip install pip~=22.0 - python -m pip install ninja cmake custatevec-cu11 pytest pytest-mock flaky pytest-cov + python -m pip install ninja cmake custatevec-cu$CUDA_MAJOR_VERSION pytest pytest-mock flaky pytest-cov - name: Build and install package env: @@ -222,7 +222,7 @@ jobs: run: | cd main rm -rf build - CMAKE_ARGS="-DPL_BACKEND=${{ matrix.pl_backend }} -DENABLE_PYTHON=ON -DCMAKE_CXX_COMPILER=$(which g++-$GCC_VERSION)" \ + CMAKE_ARGS="-DPL_BACKEND=${{ matrix.default_backend }} -DENABLE_PYTHON=ON -DCMAKE_CXX_COMPILER=$(which g++-$GCC_VERSION)" \ python -m pip install -e . -vv rm -rf build