Specify backend.

PennyLaneAI · Oct 16, 2023 · e4e24b0 · e4e24b0
1 parent 2969555
commit e4e24b0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.github/workflows/tests_linux_x86_mpich.yml b/.github/workflows/tests_linux_x86_mpich.yml
@@ -126,17 +126,16 @@ jobs:
         run: python -m pip install git+https://github.com/PennyLaneAI/pennylane.git@master
 
       - name: Build and run unit tests with MPICH backend
-        env:
-          CUQUANTUM_SDK: $(python -c "import site; print( f'{site.getsitepackages()[0]}/cuquantum/lib')")
-          PL_BACKEND: lightning_gpu
         run: |
             source /etc/profile.d/modules.sh
             module use /opt/modules/
             module load mpich
             module unload mpich
+            export CUQUANTUM_SDK=$(python -c "import site; print( f'{site.getsitepackages()[0]}/cuquantum/lib')")
             export PATH=$PATH:/usr/local/cuda/bin:/opt/mpi/mpich/include:/opt/mpi/mpich/bin
             export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/opt/mpi/mpich/lib:$CUQUANTUM_SDK
             cmake . -BBuild \
+              -DPL_BACKEND=lightning_gpu \
               -DCMAKE_PREFIX_PATH=/opt/mpi/mpich/ \
               -DENABLE_MPI=ON \
               -DCMAKE_BUILD_TYPE=RelWithDebInfo \