Error when Installing lightning-kokkos from source

PennyLane Plugins PennyLane Lightning
1

After following the instructions here: pennylane-lightning/README.rst at master · PennyLaneAI/pennylane-lightning · GitHub

I get this error when trying to test my lightning-kokkos installation:

import pennylane as qml
dev = qml.device("lightning.kokkos", wires=24)

Error:

pennylane/devices/device_constructor.py", line 260, in device
    plugin_device_class = plugin_devices[name].load()
from pennylane_lightning.core import __version__
ModuleNotFoundError: No module named 'pennylane_lightning.core'
2

Additional info:
qml.about() gives:

Platform info:           Linux-5.14.21-150500.55.49_13.0.56-cray_shasta_c-x86_64-with-glibc2.31
Python version:          3.11.10
Numpy version:           1.26.4
Scipy version:           1.14.1
Installed devices:
- default.clifford (PennyLane-0.38.0)
- default.gaussian (PennyLane-0.38.0)
- default.mixed (PennyLane-0.38.0)
- default.qubit (PennyLane-0.38.0)
- default.qubit.autograd (PennyLane-0.38.0)
- default.qubit.jax (PennyLane-0.38.0)
- default.qubit.legacy (PennyLane-0.38.0)
- default.qubit.tf (PennyLane-0.38.0)
- default.qubit.torch (PennyLane-0.38.0)
- default.qutrit (PennyLane-0.38.0)
- default.qutrit.mixed (PennyLane-0.38.0)
- default.tensor (PennyLane-0.38.0)
- null.qubit (PennyLane-0.38.0)
- lightning.qubit (PennyLane_Lightning-0.38.0)
- lightning.kokkos (PennyLane_Lightning_Kokkos-0.38.0)

I could only build for OpenMP backend, when I try building for HIP it always fails. RocM version = 6.0.3

3

Hi @Modee, welcome to the Forum!

Have you tried installing it via Spack or Docker as shown here?

If this didn’t work, could you please copy-paste here the instructions you used to install from source?

I’m thinking it might be an issue of setting CMAKE_ARGS. They should be:

CMAKE_ARGS="-DKokkos_ENABLE_HIP=ON" python -m pip install -e . --config-settings editable_mode=compat -vv

However it’s best to first try installing via Spack or Docker before trying to install from source.