quick fix

PennyLaneAI · Oct 10, 2023 · 36ea421 · 36ea421
1 parent 6f49902
commit 36ea421
Showing 1 changed file with 8 additions and 8 deletions.
diff --git 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,15 +214,15 @@ 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: 
           CUQUANTUM_SDK: $(python -c "import site; print( f'{site.getsitepackages()[0]}/cuquantum')")
         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