Skip to content

Commit 0de6132

Browse files
Lightning-Kokkos MPI installation doc update (#1349)
**Context:** In #1343 we removed the dependency for gcc-11 when installing Lightning-Kokkos/Lightning-AMDGPU. Here we update some docs instructions for installing lightning-kokkos with MPI on HPC systems as well **Description of the Change:** **Benefits:** **Possible Drawbacks:** **Related GitHub Issues:** --------- Co-authored-by: ringo-but-quantum <github-ringo-but-quantum@xanadu.ai>
1 parent 09107e1 commit 0de6132

File tree

3 files changed

+11
-8
lines changed

3 files changed

+11
-8
lines changed

.github/CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040

4141
- Remove dependency for GCC-11 when building `lightning.amdgpu`.
4242
[(#1343)](https://github.com/PennyLaneAI/pennylane-lightning/pull/1343)
43+
[(#1349)](https://github.com/PennyLaneAI/pennylane-lightning/pull/1349)
4344

4445
- Cleaned up the preprocess transforms of the lightning devices, updating the calls to `decompose` with the correct gate set.
4546
[(#1341)](https://github.com/PennyLaneAI/pennylane-lightning/pull/1341)

doc/lightning_kokkos/installation_hpc.rst

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,12 @@ We can load the following modules to enable the relevant compilers and Python en
1515
# Load the required Python and compiler modules
1616
module load cray-python
1717
module load PrgEnv-amd
18+
module load ninja cmake
19+
20+
# Create a python environment and upgrade pip
21+
python -m venv venv-pennylane
22+
source venv-pennylane/bin/activate
23+
pip install --upgrade pip
1824
1925
2026
Install Kokkos (Recommended)
@@ -23,7 +29,7 @@ Install Kokkos (Recommended)
2329
.. note::
2430

2531
Lightning-Kokkos is tested with Kokkos version 4.5.00
26-
32+
2733
We suggest first installing Kokkos with the desired configuration, following the instructions found in the Kokkos documentation.
2834
For example, the following instructions demonstrate building Kokkos for AMD MI210/250/250X GPUs:
2935

@@ -57,7 +63,7 @@ Build Kokkos for AMD GPU (``GFX90A`` architecture), and append the install locat
5763
-DKokkos_ENABLE_TESTS:BOOL=OFF \
5864
-DKokkos_ENABLE_LIBDL:BOOL=OFF
5965
cmake --build build && cmake --install build
60-
export CMAKE_PREFIX_PATH=$KOKKOS_INSTALL_PATH
66+
export CMAKE_PREFIX_PATH=$KOKKOS_INSTALL_PATH
6167
6268
6369
Install Lightning-Kokkos
@@ -88,10 +94,6 @@ Then to install Lightning-Kokkos with MPI support:
8894
# CMAKE variables for building Lightning-Kokkos with MPI
8995
export CMAKE_ARGS="-DENABLE_MPI=ON -DCMAKE_CXX_COMPILER=hipcc"
9096
91-
# Extra variables to avoid hipcc linking issues
92-
export CMAKE_ARGS="${CMAKE_ARGS} -DCMAKE_CXX_FLAGS='--gcc-install-dir=/opt/cray/pe/gcc/11.2.0/snos/lib/gcc/x86_64-suse-linux/11.2.0/'"
93-
export CMAKE_ARGS="${CMAKE_ARGS} -DCMAKE_CXX_COMPILER_CLANG_SCAN_DEPS:FILEPATH=/opt/rocm-6.2.4/lib/llvm/bin/clang-scan-deps"
94-
9597
PL_BACKEND="lightning_kokkos" python scripts/configure_pyproject_toml.py
9698
python -m pip install .
9799
@@ -116,4 +118,4 @@ To submit a job, for example on 2 nodes, the following SLURM script can be used:
116118
export HSA_ENABLE_PEER_SDMA=0
117119
118120
srun --ntasks=16 --cpus-per-task=7 --gpus-per-task=1 --gpu-bind=closest python pennylane_quantum_script.py
119-
121+

pennylane_lightning/core/_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,4 @@
1616
Version number (major.minor.patch[-label])
1717
"""
1818

19-
__version__ = "0.45.0-dev21"
19+
__version__ = "0.45.0-dev22"

0 commit comments

Comments
 (0)