Skip to content

Conversation

@renezander90
Copy link
Contributor

Contex Qrisp's qjit decorators allows running Qrisp code via the Catalyst compilation infrastructure. Thereby, the backend is qml.device("lightning.qubit", wires=0). This PR aims to facilitate using different backends, including "lightning.gpu", "lightning.amdgpu".

Description of change

Added keyword argument devicefor qrisp's qjit decorator:

from qrisp import *
from qrisp.jasp import qjit

import pennylane as qml
dev = qml.device("lightning.qubit", wires=0)

@qjit(device=dev)
def test_fun(i):
    qv = QuantumFloat(i, -2)
    with invert():
        cx(qv[0], qv[qv.size-1])
        h(qv[0])
    meas_res = measure(qv)
    return meas_res + 3

test_fun(4)

@positr0nium
Copy link
Contributor

Thanks :) Please don't forget to update the docstring 😇

@renezander90
Copy link
Contributor Author

Executing a qrisp program with qjit on "lightning.gpu" was successfully tested using NVIDIA RTX A5000.

@renezander90 renezander90 marked this pull request as ready for review January 9, 2026 20:59
@positr0nium positr0nium merged commit 3911cec into main Jan 13, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants