Merge branch 'statePrep_MPS' of github.com:PennyLaneAI/pennylane-ligh…

…tning into statePrep_MPS
PennyLaneAI · Dec 16, 2024 · b4d10d8 · b4d10d8
2 parents 98e65d9 + c3ba71d
commit b4d10d8
Show file tree

Hide file tree

Showing 6 changed files with 27 additions and 14 deletions.
diff --git a/.github/CHANGELOG.md b/.github/CHANGELOG.md
@@ -92,6 +92,9 @@
 
 ### Bug fixes
 
+* Update CUDA version to 12.4 for GPU tests on CI.
+  [(#1023)](https://github.com/PennyLaneAI/pennylane-lightning/pull/1023)
+
 * Fix Lightning Kokkos `[[maybe_unused]]` and `exp2` errors with hipcc.
   [(#1018)](https://github.com/PennyLaneAI/pennylane-lightning/pull/1018)
 

diff --git a/.github/workflows/tests_gpu_cpp.yml b/.github/workflows/tests_gpu_cpp.yml
@@ -40,14 +40,15 @@ jobs:
     strategy:
       max-parallel: 1
       matrix:
-        cuda_version: ["12"]
+        cuda_version_maj: ["12"]
+        cuda_version_min: ["4"]
 
     steps:
       - name: Validate GPU version and installed compiler
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/${{ matrix.cuda_version }}
+          module load cuda/${{ matrix.cuda_version_maj }}.${{ matrix.cuda_version_min }}
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version

diff --git a/.github/workflows/tests_gpu_python.yml b/.github/workflows/tests_gpu_python.yml
@@ -40,14 +40,15 @@ jobs:
     strategy:
       max-parallel: 1
       matrix:
-        cuda_version: ["12"]
+        cuda_version_maj: ["12"]
+        cuda_version_min: ["4"]
 
     steps:
       - name: Validate GPU version and installed compiler
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/${{ matrix.cuda_version }}
+          module load cuda/${{ matrix.cuda_version_maj }}.${{ matrix.cuda_version_min }}
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version

diff --git a/.github/workflows/tests_lkcuda_cpp.yml b/.github/workflows/tests_lkcuda_cpp.yml
@@ -48,15 +48,17 @@ jobs:
       max-parallel: 1
       matrix:
         os: [ubuntu-22.04]
-        exec_model: ["CUDA"]
         kokkos_version: ["4.5.00"]
+        exec_model: ["CUDA"]
+        cuda_version_maj: ["12"]
+        cuda_version_min: ["4"]
 
     steps:
       - name: Validate GPU version and installed compiler
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/12.2
+          module load cuda/${{ matrix.cuda_version_maj }}.${{ matrix.cuda_version_min }}
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version
@@ -110,8 +112,10 @@ jobs:
       matrix:
         os: [ubuntu-22.04]
         pl_backend: ["lightning_kokkos"]
-        exec_model: ["CUDA"]
         kokkos_version: ["4.5.00"]
+        exec_model: ["CUDA"]
+        cuda_version_maj: ["12"]
+        cuda_version_min: ["4"]
 
     name: C++ Tests (${{ matrix.pl_backend }}, kokkos-${{ matrix.kokkos_version }}, model-${{ matrix.exec_model }})
     runs-on:
@@ -124,7 +128,7 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/12.2
+          module load cuda/${{ matrix.cuda_version_maj }}.${{ matrix.cuda_version_min }}
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version
@@ -190,7 +194,7 @@ jobs:
         run: |
           cd main
           cmake . -BBuild -G Ninja \
-            -DCMAKE_BUILD_TYPE=RelWithDebInfo \
+            -DCMAKE_BUILD_TYPE=Release \
             -DBUILD_TESTS=ON \
             -DENABLE_PYTHON=OFF \
             -DSCIPY_OPENBLAS=$(python -c "import scipy_openblas32; print(scipy_openblas32.get_lib_dir())") \

diff --git a/.github/workflows/tests_lkcuda_python.yml b/.github/workflows/tests_lkcuda_python.yml
@@ -53,15 +53,17 @@ jobs:
       max-parallel: 1
       matrix:
         os: [ubuntu-22.04]
-        exec_model: ["CUDA"]
         kokkos_version: ["4.5.00"]
+        exec_model: ["CUDA"]
+        cuda_version_maj: ["12"]
+        cuda_version_min: ["4"]
 
     steps:
       - name: Validate GPU version and installed compiler
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/12.2
+          module load cuda/${{ matrix.cuda_version_maj }}.${{ matrix.cuda_version_min }}
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version
@@ -115,8 +117,10 @@ jobs:
       matrix:
         os: [ubuntu-22.04]
         pl_backend: ["lightning_kokkos", "all"]
-        exec_model: ["CUDA"]
         kokkos_version: ["4.5.00"]
+        exec_model: ["CUDA"]
+        cuda_version_maj: ["12"]
+        cuda_version_min: ["4"]
 
     name: Python Tests (${{ matrix.pl_backend }}, kokkos-${{ matrix.kokkos_version }}, model-${{ matrix.exec_model }})
     runs-on:
@@ -129,7 +133,7 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/12.2
+          module load cuda/${{ matrix.cuda_version_maj }}.${{ matrix.cuda_version_min }}
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version

diff --git a/pennylane_lightning/core/_version.py b/pennylane_lightning/core/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.40.0-dev36"
+__version__ = "0.40.0-dev38"